將多維數組列表轉換為單個多維數組？

Question

令Y為100個ndarray的列表，使得Y [i]為圖像的ndarray，其形狀為160x320x3。

我希望X不能是包含所有圖像的ndarrays，我這樣做如下：

x = [ y[i] for i in range(0,10) ]

但是它會產生100個160X320X3 ndarray的列表。 如何修改它以獲得形狀為100x160x320x3的ndarray？

Answer 1

在Y上調用np.array （即np.array(Y) ）應將ndarray列表轉換為一個ndarray，並且第一個軸的大小對應於列表的長度。

演示 ：

>>> x = np.array([[1,2], [3,4]])
>>> c = [x,x] # list of 2x2 arrays
>>> c
[array([[1, 2],
       [3, 4]]), 
array([[1, 2],
       [3, 4]])]
>>> np.array(c) # 2x2x2 array
array([[[1, 2],
        [3, 4]],

       [[1, 2],
        [3, 4]]])

只需撥打np.array上Y ，或x ，如果你只想要一片Y 。