Преобразование списка массивов NumPy в массив 5D NUMPY

Question

У меня есть база данных из 7000 объектов (list_of_objects), каждый из этих файлов содержит массив с размером 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.Как я могу сделать это вместо этого?

Answer 1

fnl_lst = np.stack([ind for ind in read_array(obj) for obj in list_of_objects])

или просто добавьте к существующему коду:

fnl_lst = np.stack(fnl_lst)

UPD: комментарием hpaulj, если my_array действительно 10x5x50x50x3, этого может быть достаточно:

fnl_lst = np.stack([read_array(obj) for obj in list_of_objects])