作者:mobiledu2502924307 | 来源:互联网 | 2023-10-10 13:08
我有两个数组my_arr和距离.例如:
my_arr= np.array([0, 1, 3, 4, 5, 3, 5])
distances = np.array([18, 47, 20, 10, 26, 22, 13])
我想得到一个索引数组,其形状是基于最小距离的np.unique(my_arr).size.所以在前面的例子中,我会得到:
# indices of my_arr
indices_of_my_arr= np.array([0, 1, 2, 3, 6])
除了循环或地图有一个聪明的方法来做到这一点?
编辑:
另一个例子:
my_arr = np.array([0, 2, 3, 1, 3, 4, 4, 5])
dist = np.array([10, 12, 15, 18, 5, 14, 45, 8])
我预计:
[0, 1, 3, 4, 5, 7]
解决方法:
你可以使用np.lexsort和np.unique –
idx = np.lexsort([distances, my_arr])
out = np.sort(idx[np.unique(my_arr[idx], return_index=1)[1]])