Question

我有两个这样的2D numpy数组，代表三点之间的x / y距离。我需要将x / y距离作为单个数组中的元组。

所以来自：

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

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

我需要：

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

我尝试过使用dstack / hstack / vstack / concatenate的各种排列，但它们似乎都没有按我的意愿行事。代码中的实际数组可能是巨大的，因此迭代python中的元素并“手动”进行重新排列不是速度方面的选择。

修改这就是我最终想出的：https://gist.github.com/807656

Answer 1

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

像你想要的那样返回dists。之后它不是“一个2元组的2D阵列”，而是一个普通的3D阵列，其中第三个轴是两个原始阵列的串联。

你看：

dists.shape # (3, 3, 2)

Answer 2

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

将两个2D numpy数组连接成一个2元组的2D数组

2 个答案: