如何在Python的单个元素中对具有不同字符优先级的嵌套列表进行排序?

时间:2013-01-04 07:45:13

标签: python list sorting nested-lists

很抱歉无法获得好的头衔。

我有一个清单:

some_list = [[X, '4x01'], [X, '3x02'], [X, '4x02'], [X, '3x01']]

X是我不在乎的事情。

我想使用以下优先级对列表中的第二个元素进行排序:

  1. 第一个字符按降序排列。
  2. 最后2个字符按升序排列。
  3. 最终输出应为:

    some_list = [[X, '4x01'], [X, '4x02'], [X, '3x01'], [X, '3x02']]
    

    注意:提供错误答案的解决方案是:

    output = sorted(some_list, key=lambda lis: lis[3], reverse=True)
    

2 个答案:

答案 0 :(得分:6)

output = sorted(some_list, key=lambda x: (-ord(x[1][0]), x[1][-2:]))

运行示例(将X定义为None,以便我可以复制粘贴列表定义):

>>> X = None
>>> some_list = [[X, '4x01'], [X, '3x02'], [X, '4x02'], [X, '3x01']]
>>> output = sorted(some_list, key=lambda x: (-ord(x[1][0]), x[1][-2:]))
>>> output
[[None, '4x01'], [None, '4x02'], [None, '3x01'], [None, '3x02']]

答案 1 :(得分:1)

在这种情况下,你可以像@Amber一样使用复合键。但是,这种方法并不总是有效:例如,如果两个组件的长度都是两个字符,那么它就会失败(因为-ord(x[1][0])技巧将不再起作用)。

这是一个更通用的解决方案:

In [15]: X = None

In [16]: some_list = [[X, '4x01'], [X, '3x02'], [X, '4x02'], [X, '3x01']]

In [17]: l = sorted(some_list, key=lambda (x,y):y[-2:])

In [18]: l = sorted(l, key=lambda (x,y):y[:1], reverse=True)

In [19]: l
Out[19]: [[None, '4x01'], [None, '4x02'], [None, '3x01'], [None, '3x02']]

它一次对列表中的一个标准进行排序,并利用Python的排序为stable这一事实。

更通用的方法是提供比较器:

def f((x1,y1), (x2,y2)):
    c = cmp(y1[:1], y2[:1])
    if c != 0: return -c
    return cmp(y1[-2:], y2[-2:])

可以使用如下:

In [39]: sorted(some_list, cmp=f)
Out[39]: [[None, '4x01'], [None, '4x02'], [None, '3x01'], [None, '3x02']]