numpy 数组排序np.sort()、np.argsort() |
您所在的位置:网站首页 › 将数组中的元素从小到大进行排序 › numpy 数组排序np.sort()、np.argsort() |
np.sort(a, axis=-1, kind=‘quicksort’, order=None) 将指定轴上的每一个元素都按照从小到大的顺序排列。 Params: axis:默认是-1,即最大维度,二维数组就是轴1,三维数组就是轴2。kind:排序规则,有三个选项,如下,使用默认的就好 kindspeedworst casework spacestable‘quicksort’1O(n^2)0no‘mergesort’2O(n*log(n))~n/2yes‘heapsort’3O(n*log(n))0no order:指定排序的规则。举例: arr = np.array([[7,3,9], [9,0,8]]) np.sort(arr,axis=0) array([[7, 0, 8], [9, 3, 9]])指定axis=0,把每一列按照从小到大排列。 np.sort(arr,axis=1) array([[3, 7, 9], [0, 8, 9]])指定axis=1,把每一行按照从小到大排列。 np.argsort(a, axis=-1, kind=‘quicksort’, order=None) 返回的是数组值从小到大排序的索引值。就是把数组升序排序,然后看看各个元素在原来数组中的索引值,看看排在第一小的数在原来数组的哪个位置。形状与传入数组相同。 参数使用同上。 举例: 这里我们举一个简单的例子,万变不离其宗。 ind = np.argsort([4,3,5,6,0],axis=0) ind array([4, 1, 0, 2, 3], dtype=int64)附图解释。每个数字排序时,它的index也随着数值走。最后返回其原来的index。 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |