Nối hai mảng có dải 2D vào một mảng 2D gồm 2 bộ tóan

Question

Tôi có hai mảng numpy 2D như thế này, đại diện cho khoảng cách x/y giữa ba điểm. Tôi cần khoảng cách x/y như tuples trong một mảng duy nhất.

Vì vậy từ:

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

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

tôi cần:

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

Tôi đã cố gắng sử dụng hoán vị khác nhau của dstack/hstack/vstack/concatenate, nhưng không ai trong số họ dường như làm những gì tôi muốn . Các mảng thực trong mã chịu trách nhiệm là khổng lồ, vì vậy việc lặp lại các phần tử trong python và thực hiện sắp xếp lại "thủ công" không phải là một lựa chọn tốc độ khôn ngoan.

Edit: Đây là những gì tôi đã đưa ra cuối cùng: https://gist.github.com/807656

Answer 1

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

lợi nhuận dists như bạn muốn họ. Sau đó nó không phải là "một mảng 2D của 2-tuples", nhưng một mảng 3D bình thường nơi trục thứ ba là nối của hai mảng ban đầu.

Bạn thấy:

dists.shape # (3, 3, 2) 

Answer 2

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