python数组第一和第二元素的紧凑形式

Question

我定义了两个数组：

a=np.array([[2,3,4],[5,6,7],[8,9,10]])
b=np.array([-1,-2])

并创建了第三个：

x=np.asarray([[x - a/2, x + a/2] for x in b])

现在，我已经定义了两个变量

u,v = x[:,0], x[:,1]

我的问题非常简单：是否有一种方法可以使用仅 数组操作来定义那些不带逗号的变量？如果我写

 u,v = x[:,]

排序错误。

Answer 1

如果x为2D：

u, v = x.T

如果x为ND：

u, v = np.swapaxes(x, 0, 1)

确认：

>>> np.all(u == x[:, 0])
True

>>> np.all(v == x[:, 1])
True