Python中的a.sort,sorted(a),np_argsort(a)和np.lexsort(b,a)

以特定顺序对数据元素进行排序是经常需要的操作。为了对数组中的元素进行排序,python使用名为sorted()和array.sort()的函数。

排序(数组)

此函数返回排序后的数组,而不修改原始数组。

a = [9,5,3,1,12,6]
b = sorted([9,5,3,1,12,6])
print "Sorted Array :\n",
print (b)
print "Original Array :\n",
print (a)

运行上面的代码给我们以下结果-

Sorted Array :
[1, 3, 5, 6, 9, 12]
Original Array :
[9, 5, 3, 1, 12, 6]

list.sort()

sort函数通过对提供的数组进行就地修改来返回已排序的数组。因此,原始数组将被修改,如下例所示。

a = [9,5,3,1,12,6]
print "Original Array :\n",
print (a)
print "Sorted Array :\n",
a.sort()
print (a)

运行上面的代码给我们以下结果-

Original Array :
[9, 5, 3, 1, 12, 6]
Sorted Array :
[1, 3, 5, 6, 9, 12]

因此,sorted()函数要慢一些,sort()因为它创建了原始数组的副本然后对其进行了修改。

使用Numpy可以完成更复杂的排序要求。Numpy是一个python库,由于它提供了许多高级功能,因此在科学数据处理中被广泛使用。在以下示例中,我们将看到纯python排序方法和Numpy排序方法。

numpy.argsort

numpy中的此函数返回排序数组的索引,而不是数组元素的索引。在下面的示例中,我们采用数组,打印其元素以及每个元素的索引。然后我们应用argsort函数,该函数为我们提供排序后的数组的索引,结果也是一个数组。

import numpy as np
x = np.array([9,5,3,1,12,6])
print(x)

#Print the positions of elements
for i in range(len(x)):
print "[",i,"]",x[i],
print "\n"
# Print the indices of sorted elements
s = np.argsort(x)
print(s)

运行上面的代码给我们以下结果-

[ 9 5 3 1 12 6]

[ 0 ] 9 [ 1 ] 5 [ 2 ] 3 [ 3 ] 1 [ 4 ] 12 [ 5 ] 6

[3 2 1 5 0 4]

numpy.lexsort

此功能用于使用涉及多个数组的多个排序键进行排序。例如,我们首先对A列中的数据进行排序,然后对B列中的值进行排序。在下面的示例中,我们采用表示A列和B列的两个数组。在应用lexsort()函数时,首先按A列然后按B列进行排序排序结果为包含A列中元素索引的数组。

import numpy as np
colA = [2,5,1,8,1] # First column
colB = [9,0,3,2,0] # Second column
# Sort by ColA and then by colB
sorted_index = np.lexsort((colB,colA))
print(sorted_index)
#print the result showing the
#column values as pairs
print [(colA[i],colB[i]) for i in sorted_index]

运行上面的代码给我们以下结果-

[4 2 0 1 3]
[(1, 0), (1, 3), (2, 9), (5, 0), (8, 2)]

如您所见,colA中最低的两个值分别在索引位置2和4处为1和1。但是结果显示4和2为升序,因为B列中分别为0和3的值也被排在首位0然后3,结果为4和2。