我对python有点新鲜。我有一个名为featureExtraction的函数,它返回一个图像的一维数组。我需要逐行堆叠所有这样的1-d数组以形成2-d数组。我在MATLAB中有以下等效代码。
I1=imresize(I,[256 256]);
Features(k,:) = featureextraction(I1);
featureextraction返回1-d行向量,该向量逐行堆叠以形成2-d数组。 python中等效的代码片段是什么?
提前谢谢你。
答案 0 :(得分:6)
不确定您要找的是什么,但也许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]])
答案 1 :(得分:4)
您可以使用numpy.vstack()
:
a = np.array([1,2,3])
np.vstack((a,a,a))
#array([[1, 2, 3],
# [1, 2, 3],
# [1, 2, 3]])