python将三维数组保存,如何在Python / numpy中将3D数组结果保存到4D数组？

I am reading the information about 32 X 32 RGB images. So it is a 3D array with shape (32,32,3) third dimension holds the color R, G and B

Now, I want to read 50 such images and make a array of these images. So I decide to make a 4D array which will have dimension (50, 32, 32, 3) here 50 in first dimension is the number of images and the second, third and fourth dimension are the dimension of the image which are (32, 32, 3)

I tried doing it using concatenate but I get errors. Is there any way to do this?

解决方案

You need to add an axis before concatenation, e.g.

import numpy as np

arrs = [np.random.random((32, 32, 3))

for i in range(50)]

res = np.concatenate([arr[np.newaxis] for arr in arrs])

res.shape

# (50, 32, 32, 3)

Edit: alternatively, in this case you can simply call np.array on your list of arrays:

res = np.array(arrs)

res.shape

# (50, 32, 32, 3)