Numpy提供了大量用数组操作的函数,其中不乏常见的排序函数。
这里讲下numpy.sort、numpy.argsort、numpy.lexsort三种排序函数的用法。
1、如何对数组元素进行快速排序?
使用numpy.sort函数可以对数组进行排序,并返回排序好的数组。
使用方法:
numpy.sort(a, axis=-1, kind=None, order=None)
参数:
a : 要排序的数组;
axis :按什么轴进行排序,默认按最后一个轴进行排序;
kind :排序方法,默认是快速排序;
order : 当数组定义了字段属性时,可以按照某个属性进行排序;
import numpy as np
# 创建一个一维数组
x1 = np.array([1,8,2,4])
x1
'''
一维数组:
array([1, 8, 2, 4])
'''
# 排序
np.sort(x1)
'''
输出:
array([1, 2, 4, 8])
'''
import numpy as np
# 创建一个二维数组
x2 = np.array([[1,8,2,4],[4,5,1,3]])
x2
'''