我想在 Python 中根据预先排序的列表对列表进行排序
presorted_list = ['2C','3C','4C','2D','3D','4D']
unsorted_list = ['3D','2C','4D','2D']
尽管并非所有元素都出现在未排序的列表中，但有没有办法对列表进行排序以反映预排序的列表？
我希望结果看起来像这样：
after_sort = ['2C','2D','3D','4D']
谢谢！
【问题讨论】：
In [5]: sorted(unsorted_list, key=presorted_list.index)
Out[5]: ['2C', '2D', '3D', '4D']
或者，为了获得更好的性能（特别是当len(presorted_list) 很大时），
In [6]: order = {item:i for i, item in enumerate(presorted_list)}
In [7]: sorted(unsorted_list, key=order.__getitem__)
Out[7]: ['2C', '2D', '3D', '4D']
有关如何使用keys 进行排序的更多信息，请参阅优秀的Howto Sort wiki。
如果unsorted_list 包含不在presorted_list 中的项目（例如'6D'），则上述方法将引发错误。您首先必须决定如何对这些项目进行排序。如果您希望将它们放在列表的末尾，您可以使用
In [10]: unsorted_list = ['3D','2C','6D','4D','2D']
In [11]: sorted(unsorted_list, key=lambda x: order.get(x, float('inf')))
Out[11]: ['2C', '2D', '3D', '4D', '6D']
或者如果您希望将此类项目放在列表的前面，请使用
In [12]: sorted(unsorted_list, key=lambda x: order.get(x, -1))
Out[12]: ['6D', '2C', '2D', '3D', '4D']
【讨论】：
.index 将引发值错误。
presorted_list 中的项目，请使用sorted(itertools.ifilter(lambda x: x in presorted_list, unsorted_list), key=order.__getitem__) 之类的内容。