arrays - 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 是二维:

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