python - 使用引用顺序的自定义排序列表

Question

我想按每行第一个索引 0 中的值对数组进行排序。该数组具有以下形式:

['AAA', 1, 2, -1]
['BB+', 3, 2, 1]
['AA-', 5, 4, 1]
['AA+', 4, 2, 2]

正如您可能看到的那样，我想按一些信用评级进行排序。不幸的是，仅使用内置函数进行排序并不能提供正确的顺序。我想做的是以某种方式指定顺序，例如:

correct_order = ['AAA', 'AA+',...,'BBB+', 'BBB', 'BB+',...]

我可以想象有无数种方法可以实现这一点，可能没有指定它。我不担心性能，因为我永远不会处理超过 22 行。

需要强调的重要一点是，排序将基于每行中的索引 0，但保留数组的其余部分。此外，不能保证数组将包含 correct_order 列表中的每个“评级”。

我曾在论坛上看到过类似问题的解释，但无法将它们适应我的数据集。非常感谢任何有关最简单解决方案的帮助。

Answer 1

作为您的键，您可以在正确排序的评级字符串列表中使用评级字符串的索引:

>>> correct_order = ['AAA', 'AA+', 'BBB+', 'BBB', 'BB+']
>>> list(sorted("BBB BBB+ BB+ AAA AA+ BBB+ BBB AA+".split(), key=lambda s:correct_order.index(s)))
['AAA', 'AA+', 'AA+', 'BBB+', 'BBB+', 'BBB', 'BBB', 'BB+']

使其适应您的特定数据集(评级字符串位于列表列表中每个列表的第一个元素中):

correct_order = ['AAA', 'AA+',...,'BBB+', 'BBB', 'BB+',...]
the_lists.sort(key=lambda l: correct_order.index(l[0]))