Python - 如何使用用户定义的函数执行间接排序？

Question

numpy.argsort返回一个排序列表来执行间接排序，但它似乎并不接受用户定义的函数来比较两个元素。

我想知道如何根据与用户定义函数的比较得出排序列表。

在我的情况，我有结果表：

a = [[1,2,3,5,6,7,8],[4,6,2,5,6,3,4],...] 

而我的函数来比较是：

def argMedian(A): 
    s = np.array([sum(A[:i+1]) for i in range(Nmentions)]) 
    mid = float(s[Nmentions-1])/2 
    return np.argwhere(mid < s)[0][0] 

def tieBreaking(A, B): 
Ac = A 
Bc = B 
medA = argMedian(Ac) 
medB = argMedian(Bc) 
while medA == medB: 
    Ac[medA] -= 1 
    Bc[medB] -= 1 
    medA = argMedian(Ac) 
    medB = argMedian(Bc) 
return 1 if medA > medB else -1 

这是majority judgment的实现。列表包含每个年级的选票数量。如果两个列表的中位数相等，则两个列表的中位投票都将被删除，中位数的比较将再次进行测试。在这里，我认为索引0是最好的等级，索引7是最差的。

我需要执行间接排序，因为我想知道排名。

Answer 1

对于基本的Python计算间接排序，一些小技巧存在。他们在Equivalent of Numpy.argsort() in basic python?进行了比较。

以最快的解决方案，可以添加比较功能。

在Python 2.7：

sorted(range(len(results)), cmp=tieBreaking, key=results.__getitem__) 

在Python 3，应注意使cmp已经消失，但功能cmp_to_key可以取代它。

Answer 2

您不需要使用numpy对列表中的元素进行排序。您可以对列表或排序（）进行排序（）操作。如果需要，定义您的自定义比较器。

Answer 3

如果您使用Python> = 3.4，则可以使用statistics.median_low()。

from random import randrange 
from statistics import median_low 

a = [[randrange(8) for _ in range(7)] for _ in range(10)] 

print("unsorted") 
for item in a: 
    print(item) 

a.sort(key=median_low) 

print("\nsorted") 
for item in a: 
    print(item) 

输出：

unsorted 
[4, 2, 2, 1, 4, 7, 4] 
[2, 2, 2, 7, 5, 5, 6] 
[3, 0, 0, 5, 5, 3, 1] 
[4, 7, 6, 2, 6, 7, 3] 
[4, 7, 7, 1, 2, 7, 7] 
[6, 2, 6, 5, 6, 7, 2] 
[7, 1, 6, 0, 0, 7, 1] 
[0, 5, 1, 2, 1, 7, 7] 
[2, 7, 6, 7, 5, 4, 7] 
[6, 5, 2, 3, 5, 0, 3] 

sorted 
[7, 1, 6, 0, 0, 7, 1] 
[0, 5, 1, 2, 1, 7, 7] 
[3, 0, 0, 5, 5, 3, 1] 
[6, 5, 2, 3, 5, 0, 3] 
[4, 2, 2, 1, 4, 7, 4] 
[2, 2, 2, 7, 5, 5, 6] 
[4, 7, 6, 2, 6, 7, 3] 
[6, 2, 6, 5, 6, 7, 2] 
[2, 7, 6, 7, 5, 4, 7] 
[4, 7, 7, 1, 2, 7, 7] 

EDIT充分替代的解决方案：

考虑排序的列表如下： 1.查找列表的median_low并将其移动到的前名单 2.查找列表[1：]的median_low并将其移至第二个地点 3.查找列表[2：]的中位数下降...

您可以按中位数值对原始列表进行排序，也可以创建按中位值对元素进行排序的键。

def def keyfunc(x): 
    t = x[:] 
    return [t.pop(t.index(median_low(t))) for _ in range(len(t))] 

a = [ 
    [4, 2, 2, 1, 4, 7, 4], 
    [2, 2, 2, 7, 5, 5, 6], 
    [3, 0, 0, 5, 5, 3, 1], 
    [7, 1, 6, 0, 0, 6, 1], # tie for first four rounds, but then wins 
    [4, 7, 6, 2, 6, 7, 3], 
    [6, 2, 6, 5, 6, 7, 2], 
    [7, 1, 6, 0, 0, 7, 1], # tie for first four rounds 
    [0, 5, 1, 2, 1, 7, 7], 
    [2, 7, 6, 7, 5, 4, 7], 
    [6, 5, 2, 3, 5, 0, 3] 
] 

a.sort(key=keyfunc) 

print("\nsorted") 
for item in a: 
    print(item) 

输出：

sorted 
[7, 1, 6, 0, 0, 6, 1] 
[7, 1, 6, 0, 0, 7, 1] 
[0, 5, 1, 2, 1, 7, 7] 
[3, 0, 0, 5, 5, 3, 1] 
[6, 5, 2, 3, 5, 0, 3] 
[4, 2, 2, 1, 4, 7, 4] 
[2, 2, 2, 7, 5, 5, 6] 
[4, 7, 6, 2, 6, 7, 3] 
[6, 2, 6, 5, 6, 7, 2] 
[2, 7, 6, 7, 5, 4, 7]