numpy.argsort()

先对数组从小到大排序再返回对应的下标

from numpy import argsort, array
a = array([4, 1, 3, 2])
print(argsort(a))
"""output
[1 3 2 0] """
发布了19 篇原创文章 · 获赞 3 · 访问量 724

猜你喜欢

转载自blog.csdn.net/weixin_43486780/article/details/104531551