Компактная форма 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]

MyВопрос предельно прост: есть ли способ определить эти переменные без запятой, используя only операции с массивами?Если я напишу

 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