У меня есть список двухмерных массивов. Как преобразовать список в трехмерный массив numpy?

Question

L - это список из 1000 массивов размером (300,300). Поэтому мне нужно преобразовать L в массив numpy (3D) размером (300,300) x 1000.

Answer 1

Используйте np.stack. Для этого требуется последовательность (у вас есть список, который будет работать) массивов для укладки вдоль новой оси. Обратите внимание, что axis=-1 дает вам спецификацию оси, которую вы просили.

# list of arrays made of random noise
x_list = [np.random.normal(0.0, 1.0, size=(300, 300)) for _ in range(1000)]
# array made of arrays in list
x_array = np.stack(x_list, axis=-1)

print("Shape of array: ", x_array.shape)

Это дает

>>> Shape of array: (300, 300, 1000)