Łączenie dwóch tablic 2D numpy w pojedynczą tablicę 2D z 2-krotkami

Question

Mam dwie taktyczne liczby 2D, reprezentujące odległości x/y pomiędzy trzema punktami. Potrzebuję odległości x/y jako krotek w jednej tablicy.

Więc od:

x_dists = array([[ 0, -1, -2], 
       [ 1, 0, -1], 
       [ 2, 1, 0]]) 

y_dists = array([[ 0, -1, -2], 
       [ 1, 0, -1], 
       [ 2, 1, 0]]) 

muszę:

dists = array([[[ 0, 0], [-1, -1], [-2, -2]], 
       [[ 1, 1], [ 0, 0], [-1, -1]], 
       [[ 2, 2], [ 1, 1], [ 0, 0]]]) 

Próbowałem przy użyciu różnych permutacji dstack/hstack/vstack/concatenate, ale żaden z nich nie wydają się zrobić to, co chcę . Rzeczywiste tablice w kodzie mogą być gigantyczne, więc iteracja nad elementami w pythonie i wykonanie "ręcznego" przegrupowania nie jest opcją na szybkość.

Edit: To co wymyśliłem w końcu: https://gist.github.com/807656

Answer 1

import numpy as np 
dists = np.vstack(([x_dists.T], [y_dists.T])).T 

powraca dists jakby je chciał. Później nie jest to "pojedyncza tablica 2D z 2 krotkami", ale normalna tablica 3D, gdzie trzecia oś jest połączeniem dwóch oryginalnych tablic.

Zobaczysz:

dists.shape # (3, 3, 2) 

Answer 2

numpy.rec.fromarrays([x_dists, y_dists], names='x,y')