在numpy 2维数组中堆叠列以获取1维数组

时间:2018-09-03 17:15:26

标签: python arrays numpy

如何将二维Numpy数组中的所有列堆叠为一维数组。

即我有:

x = np.array([[1, 3, 5],[2, 4, 6]])

我想得到:

np.array([1, 2, 3, 4, 5, 6])

有没有没有循环或列表理解的方法?

2 个答案:

答案 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])