Преобразование списка массивов numpy в массив 5D numpy
У меня есть база данных из 7000 объектов (list_of_objects), каждый из этих файлов содержит массив numpy размером 10x5x50x50x3. Я хотел бы создать массив 5d numpy, который будет содержать 7000*10x5x50x50x3. Я попытался сделать это, используя два цикла for. Мой пример кода:
fnl_lst = []
for object in list_of_objects:
my_array = read_array(object) # size 10x5x50x50x3
for ind in my_array:
fnl_lst.append(ind)
fnl_lst= np.asarray( fnl_lst) # print(fnl_lst) -> (70000,)
Результатом этого кода является вложенный массив numpy, который содержит 70000 массивов, каждый из которых имеет размер 5x50x50x3. Однако вместо этого я хотел бы построить массив 5d с размером 70000x5x50x50x3. Как я могу сделать это вместо этого?
Вы уверены, что все ind одинаковой формы? Что производит np.vstack([read_array(o) for o in list_of_objects])?
@hpaulj Я перепроверю. Может быть, это моя проблема, что ind по какой-то причине не имеют одинакового размера.
Что такое fnl_lst.dtype? Если object, то да, есть некоторые различия в формах.
Правда один файл имеет разный размер и не смог его отследить. Теперь он работает правильно.
Ответы 1
fnl_lst = np.stack([ind for ind in read_array(obj) for obj in list_of_objects])
или просто добавьте к существующему коду:
fnl_lst = np.stack(fnl_lst)
УПД: по комментарию hpaulj, если my_array действительно 10x5x50x50x3, этого может быть достаточно:
fnl_lst = np.stack([read_array(obj) for obj in list_of_objects])
Возможный дубликат Как преобразовать список массивов numpy в один массив numpy?