將從函數返回的多個1D數組合並到2D數組python中

Question

我對python有點新鮮。 我有一個名為featureExtraction的函數，它返回一個圖像的一維數組。 我需要逐行堆疊所有這樣的1-d數組以形成2-d數組。 我在MATLAB中有以下等效代碼。

    I1=imresize(I,[256 256]);
    Features(k,:) = featureextraction(I1);

featureextraction返回1-d行向量，該向量逐行堆疊以形成2-d數組。 python中的等效代碼片段是什么？

先感謝您。

Answer 1

不知道你要找的東西，但也許vstack或column_stack ？

>>> np.vstack((a,a,a))
array([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])
>>> np.column_stack((a,a,a))
array([[0, 0, 0],
       [1, 1, 1],
       [2, 2, 2],
       [3, 3, 3],
       [4, 4, 4],
       [5, 5, 5],
       [6, 6, 6],
       [7, 7, 7],
       [8, 8, 8],
       [9, 9, 9]])

甚至只是np.array ：

>>> np.array([a,a,a])
array([[0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9],
       [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]])

Answer 2

你可以使用numpy.vstack() ：

a = np.array([1,2,3])

np.vstack((a,a,a))
#array([[1, 2, 3],
#       [1, 2, 3],
#       [1, 2, 3]])