NumPy提供了多种方法对数组元素进行排序,常用的方法有:
- 使用
numpy.sort()
函数对数组进行排序,该函数返回数组的排序副本,原数组不变。示例代码如下:
import numpy as np arr = np.array([3, 1, 2, 5, 4]) sorted_arr = np.sort(arr) print(sorted_arr)
- 使用
numpy.argsort()
函数返回数组排序后的索引值,可以根据这些索引值重新排列数组元素。示例代码如下:
import numpy as np arr = np.array([3, 1, 2, 5, 4]) indices = np.argsort(arr) sorted_arr = arr[indices] print(sorted_arr)
- 使用
numpy.lexsort()
函数对多个数组进行排序,按照最后一个数组的顺序对前面的数组进行排序。示例代码如下:
import numpy as np arr1 = np.array([3, 1, 2, 5, 4]) arr2 = np.array([9, 8, 7, 6, 5]) indices = np.lexsort((arr2, arr1)) sorted_arr1 = arr1[indices] sorted_arr2 = arr2[indices] print(sorted_arr1) print(sorted_arr2)
除了以上方法,NumPy还提供了其他排序函数,比如numpy.sort_complex()
用于对复数数组进行排序,numpy.argsort()
函数可以指定排序的轴等。具体可以查阅NumPy官方文档获取更多信息。