具有两个条件的列表中的最大值

Question

我在 Python 中有一个列表，其中每个元素都是这样的元组：

(attr1, attr2, attr3)

我想找到attr2最大但attr3 >= 100的元组。

什么是pythonic方法？

Answer 1

您必须过滤并使用 max 的key参数：

from operator import itemgetter

max(filter(lambda a: a[2] >= 100, yourlist), key=itemgetter(1))

过滤器也可以表示为生成器表达式：

max((t for t in yourlist if t[2] >= 100), key=itemgetter(1))

演示：

>>> yourlist = [(1, 2, 300), (2, 3, 400), (3, 6, 50)]
>>> max((t for t in yourlist if t[2] >= 100), key=itemgetter(1))
(2, 3, 400)
>>> max(filter(lambda a: a[2] >= 100, yourlist), key=itemgetter(1))
(2, 3, 400)

请注意，因为您进行过滤，所以很容易得到一个空列表来从中选择最大值，因此您可能需要捕获ValueError s，除非您需要该异常向上传播调用堆栈：

try:
    return max(filter(lambda a: a[2] >= 100, yourlist), key=itemgetter(1))
except ValueError:
    # Return a default
    return (0, 0, 0)

Answer 2

如果你还想要最大值的索引，你可以这样做：

 max(filter(lambda a: a[1][2] >= 100, enumerate(yourlist)), key=lambda x:x[1][1])` 

虽然我不得不承认它有点难以理解。