python中np.argsort(一维数组）用法解释

作者：羊村懒王 | 2024-05-30 22:11:44

踩

np.argsort

np.argsort 是 NumPy 中用于返回数组排序索引的函数。当你传递一个一维数组给 np.argsort 时，它会返回一个按照升序排列原始数组索引的数组。

import numpy as np

arr = np.array([5, 2, 8, 1, 7])

sorted_indices = np.argsort(arr)

print(sorted_indices)

在这个例子中，arr 是 [5, 2, 8, 1, 7]。np.argsort(arr) 返回的是排序后的索引 [3, 1, 0, 4, 2]。这表示原始数组中最小的元素是在索引 3 处，其次是索引 1 处，以此类推。

可以使用这些索引来获取排序后的数组：

sorted_arr = arr[sorted_indices]

print(sorted_arr)

这会输出 [1, 2, 5, 7, 8]，即原始数组的升序排列。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/羊村懒王/article/detail/648828