python - 在 Python 中的同一迭代中查找最小值和最大值

Question

我有以下代码:

max=max(len(j.links) for j in roads.junctions())
min=min(len(j.links) for j in roads.junctions())

roads.junctions() 返回一个包含近 100 万个路口的列表。 有没有办法在一行中找到同一迭代中项目的最小值和最大值(而不是编写一个计算它们的函数)？

Answer 1

您无法在一行中同时获得最小值和最大值，但您可以通过一个简单的循环获得它们:

min_value, max_value = float('inf'), float('-inf')
for j in roads.junctions():
    value = len(j.links)
    if value < min_value:
        min_value = value
    if value > max_value:
        max_value = value

这会产生与 min() 和 max() 相同复杂度的值:O(N)，因此是线性复杂度。它还会单独查看每个值，不需要所有值都在内存中的单个列表中。

当然你总是可以把它包装成一个函数:

def min_and_max(iterable):
    min_value, max_value = float('inf'), float('-inf')
    for value in iterable:
        if value < min_value:
            min_value = value
        if value > max_value:
            max_value = value
    return min_value, max_value

min_value, max_value = min_and_max(len(j.links) for j in roads.junctions())