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官方文檔獲取更多信息。