如何将二维Numpy数组中的所有列堆叠为一维数组。
即我有:
x = np.array([[1, 3, 5],[2, 4, 6]])
我想得到:
np.array([1, 2, 3, 4, 5, 6])
有没有没有循环或列表理解的方法?
答案 0 :(得分:4)
您可以使用ravel
:
x = np.array([[1, 3, 5],[2, 4, 6]])
res = x.ravel('F') # or x.T.ravel()
# array([1, 2, 3, 4, 5, 6])
答案 1 :(得分:4)
将flatten
与'F'
一起使用
x.flatten('F')
Out[114]: array([1, 2, 3, 4, 5, 6])