python argsort

作者：堕魂灭天发布日期:2026-02-06 浏览:72

import numpy as np

# 创建一个示例数组
arr = np.array([3, 1, 2])

# 使用 argsort 获取排序后的索引
sorted_indices = np.argsort(arr)

print("原始数组:", arr)
print("排序后的索引:", sorted_indices)
print("根据排序后的索引获取排序后的数组:", arr[sorted_indices])

解释说明：

np.argsort(arr) 返回的是数组 arr 中元素按升序排列后的索引。
在上面的例子中，arr 是 [3, 1, 2]，经过 argsort 后得到的索引是 [1, 2, 0]，表示原数组中的第 1 个元素（即 1）是最小的，第 2 个元素（即 2）是第二小的，第 0 个元素（即 3）是最大的。
最后通过 arr[sorted_indices] 可以得到排序后的数组 [1, 2, 3]。

上一篇：python的random用法

下一篇：python tkinter text