如何将数组的列拆分为三个数组 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]])
现在数据是
[[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])
最佳答案
转置，然后解压:
>>> x, y, z = data.T
>>> x
array([1, 4, 7])
关于python - 轻松拆分 numpy 数组的列，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/30820962/