python - 获取排序的索引列表，用于按给定键排序的字典列表

Question

我有一个包含多个键的字典列表，我想获得多个列表，每个列表都包含字典的索引，并按特定键排序。

例如，我有以下列表

a = [{"name": "Zoe", "age": 13}, {"name": "Adam", "age": 31}]

我知道我可以做到以下几点

from operator import itemgetter
sorted(a, key=itemgetter("name"))

获取以下信息

[{'name': 'Adam', 'age': 31}, {'name': 'Zoe', 'age': 13}]

但我真的希望能够获得两个带有索引的列表:

[1, 0] # order of the elements in `a` as sorted by name
[0, 1] # order of the elements in `a` as sorted by age

我的真实字典有更多的键值对，我返回 a 和附加列表的索引按不同的键排序而不是 a 的多个排序列表更有效。

Answer 1

您可以通过生成 range() 生成一个排序的索引列表，然后调整排序键以将索引转换为字典中的特定值a 列表:

sorted(range(len(a)), key=lambda i: a[i]['name'])
sorted(range(len(a)), key=lambda i: a[i]['age'])

如果您预先知道键，只需循环并生成多个列表；您也许可以创建一个由每个排序键键入的字典:

{key: sorted(range(len(a)), key=lambda i: a[i][key]) for key in ('name', 'age')}

然后，上面的字典理解使您可以单独根据排序键访问每个排序列表:

>>> a = [{'name': "Zoe", 'age': 13}, {'name': "Adam", 'age': 31}]
>>> {key: sorted(range(len(a)), key=lambda i: a[i][key]) for key in ('name', 'age')}
{'age': [0, 1], 'name': [1, 0]}