Python字典列表，获取最大值索引

Question

我正在尝试在字典列表中获得具有最大值'size'的字典索引，如下所示：

ld = [{'prop': 'foo', 'size': 100}, {'prop': 'boo', 'size': 200}]

使用以下代码，我可以采用最大大小：

items = [x['size'] for x in ld]
print(max(items))

我现在如何获取其索引？有没有简单的方法？

试验：

我只是想我可以做到这一点：

items = [x['size'] for x in ld]
max_val = max(items)
print(items.index(max_val))

这是正确的吗？

Answer 1

告诉max()如何计算索引序列的最大值：

max(range(len(ld)), key=lambda index: ld[index]['size'])

这将返回size键为最高的索引：

>>> ld = [{'prop': 'foo', 'size': 100}, {'prop': 'boo', 'size': 200}]
>>> max(range(len(ld)), key=lambda index: ld[index]['size'])
1
>>> ld[1]
{'size': 200, 'prop': 'boo'}

如果您一直想要那本词典，那么您可以使用：

max(ld, key=lambda d: d['size'])

并同时获得索引和字典，您可以在此处使用enumerate()：

max(enumerate(ld), key=lambda item: item[1]['size'])

更多演示：

>>> max(ld, key=lambda d: d['size'])
{'size': 200, 'prop': 'boo'}
>>> max(enumerate(ld), key=lambda item: item[1]['size'])
(1, {'size': 200, 'prop': 'boo'})

key函数将依次传递输入序列中的每个元素，max()将选择该key函数的返回值最高的元素。

使用单独的列表提取所有size值，然后将其映射回原始列表不是很有效（您现在需要遍历该列表两次）。 list.index()无法工作，因为它必须匹配整个字典，而不仅仅是其中一个值。

Answer 2

您可以使用适当的键将enumerate(ld)传递到max功能：

>>> max(enumerate(ld),key=lambda arg:arg[1]['size'])[0]
1

如果只希望字典具有最大size值，则可以使用operator.itemgetter函数作为键：

In [10]: from operator import itemgetter 

In [11]: ld = [{'prop': 'foo', 'size': 100}, {'prop': 'boo', 'size': 200}]

In [12]: fn = itemgetter('size')

In [13]: max(ld, key=fn)
Out[13]: {'prop': 'boo', 'size': 200}