Python中是否支持‘函数推导式’？能否在列表推导时直接计算均值？

1. 有没有“函数推导式”？ #

Python里根本没有*“函数推导式”*这个官方语法，常用的推导式类型包括：

• 列表推导式（就是你示例里用的[val[0] for val in data if val[1] == 0]）
• 字典推导式：{key_expr: val_expr for ...}
• 集合推导式：{expr for ...}
• 生成器表达式：(expr for ...)——惰性求值，无需一次性加载所有元素到内存

2. 能不能直接计算平均值，无需先构建列表？ #

当然可以，这种方式更高效（尤其是处理大数据集时），能避免占用额外内存存储中间列表。不过要注意处理筛选结果为空的情况，防止触发除以0的错误。

方法一：用itertools.tee拆分生成器（无中间列表）

生成器只能遍历一次，我们可以用itertools.tee复制出两个独立的生成器，分别计算总和与计数：

from itertools import tee

data = [[1, 0], [0, 1], [1, 1]]
# 创建筛选后的生成器
filtered = (val[0] for val in data if val[1] == 0)
# 拆分成两个可独立遍历的生成器
sum_gen, count_gen = tee(filtered)

total = sum(sum_gen)
count = sum(1 for _ in count_gen)

# 处理空筛选结果的边界情况
average = total / count if count != 0 else 0.0

方法二：一次遍历同时累加（更直观，无需导入模块）

直接通过循环一次遍历完成总和与计数的累加，代码逻辑更清晰：

total = 0
count = 0
for val in data:
    if val[1] == 0:
        total += val[0]
        count += 1

average = total / count if count != 0 else 0.0

这两种方法都是O(n)时间复杂度，且不会像原代码那样创建额外列表，内存占用更低。

内容的提问来源于stack exchange，提问作者Xanthan