使用numpy从列的2d数组中获取1d数组

时间:2015-10-20 12:56:24

标签: python numpy

我想使用带有列的numpy从2d数组中做1d数组,但是使用ravel我只能得到以下内容:

In [1]: import numpy as np

In [2]: a = np.arange(12)

In [3]: a = a.reshape(3,4)

In [4]: a
Out[4]:
array([[ 0,  1,  2,  3],
       [ 4,  5,  6,  7],
       [ 8,  9, 10, 11]])

In [5]: a.ravel()
Out[5]: array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11])

但我想得到:

array([ 0,  4,  8,  1,  5,  9,  2,  6, 10,  3,  7, 11])

1 个答案:

答案 0 :(得分:1)

已经找到解决方案np.flatten,只需要使用标志'F'

In [15]: a.ravel('F')
Out[15]: array([ 0,  4,  8,  1,  5,  9,  2,  6, 10,  3,  7, 11])