gpt4 book ai didi

python - 获取已排序 Numpy 数组的原始索引

转载 作者:IT老高 更新时间:2023-10-28 21:49:34 25 4
gpt4 key购买 nike

我有一个距离数组 a = np.array([20.5 ,5.3 ,60.7 ,3.0 ], 'double') 我需要排序数组的索引(例如 [3, 1, 0, 2],对于 a.sort())。 Numpy 中有一个函数可以做到这一点吗?

最佳答案

是的,有 x = numpy.argsort(a) 函数或 x = numpy.ndarray.argsort(a) 方法。它完全符合您的要求。您还可以将 argsort 作为 ndarray 对象上的方法调用,如下所示:a.argsort()

这里是文档的链接:http://docs.scipy.org/doc/numpy/reference/generated/numpy.argsort.html#numpy.argsort

关于python - 获取已排序 Numpy 数组的原始索引,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/7341557/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com