在NumPy中,argsort函数在数组中返回每个元素的索引,以按升序排列的顺序。array.argsort返回的值与array的维度相同,因此我们可以使用此索引数组扩展原始array。
以下是一个示例代码,展示了如何使用argsort和扩展索引数组:
import numpy as np
x = np.array([1, 4, 2, 3, 5]) print("原始数组:", x)
idx = np.argsort(x) print("索引数组:", idx)
result = x[idx] print("扩展结果:", result)
输出: 原始数组: [1 4 2 3 5] 索引数组: [0 2 3 1 4] 扩展结果: [1 2 3 4 5]