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