轻松拆分numpy数组的列

Question

如何将数组的列拆分为三个数组x,y,z,而无需手动分别编写每个数组[:,0],[:,1],[:,2]？

例

# Create example np array
import numpy as np
data = np.array([[1,2,3],[4,5,6],[7,8,9]])

现在data是

[[1 2 3]
 [4 5 6]
 [7 8 9]]

我想做的事:

x, y, z = data[:,0], data[:,1], data[:,2] ## Help me here!
print(x)

通缉输出:

array([1, 4, 7])

Answer 1

移调,然后打开包装:

>>> x, y, z = data.T
>>> x
array([1, 4, 7])

Answer 2

你不需要切片。

>>> import numpy as np
>>> data = np.array([[1,2,3],[4,5,6],[7,8,9]])
>>> x, y, z = data.T
>>> x
array([1, 4, 7])
>>> y
array([2, 5, 8])
>>> z
array([3, 6, 9])