python - 我如何使用附加属性扩展我的 2 个暗淡 numpy 数组

Question

我有一个 2 个暗淡的 numpy 数组，其中包含从一个起点到每个端点的距离，我想对距离进行排序，但要做到这一点，我必须为距离提供一个索引，这样我就不会丢失对端点的引用。

如何添加索引？最好没有循环

我已经使用 numpy.expand_dims 将数组扩展了 1 个维度，但我不知道如何附加索引。

所以我目前的方法是:

indexArray = [1,2,3]
distances = np.array([[0.3, 0.5, 0.2], [0.7, 0.1, 0.5], [0.2, 0.3, 0.8]])
distances = np.expand_dims(distances, axis=2)

现在距离看起来像:

[[[0.3], [0.5], [0.2]], 
 [[0.7], [0.1], [0.5]],
 [[0.2], [0.3], [0.8]]]

现在我想附加indexArray，使距离数组看起来像:

[[[1, 0.3], [2, 0.5], [3, 0.2]],
 [[1, 0.7], [2, 0.1], [3, 0.5]],
 [[1, 0.2], [2, 0.3], [3, 0.8]]]

Answer 1

您可以在这里使用切片分配，不需要 expand_dims，广播将处理其余的事情。

<小时/>

out = np.empty(distances.shape + (2,))
out[..., 0] = indexArray
out[..., 1] = distances

array([[[1. , 0.3],   
        [2. , 0.5],   
        [3. , 0.2]],  

       [[1. , 0.7],   
        [2. , 0.1],   
        [3. , 0.5]],  

       [[1. , 0.2],   
        [2. , 0.3],   
        [3. , 0.8]]])