如何将含if判断的Datetime遍历for循环转为列表推导式

你遇到的问题很典型——很多刚接触列表推导式的同学会误以为它能替代所有for循环，但其实列表推导式的核心用途是生成新列表，而不是执行带有副作用的操作（比如修改已有字典的计数）。

先回顾一下你原来的循环代码，逻辑是完全没问题的：

# 创建存储结果的字典
trip_counts = {'AM': 0, 'PM': 0}
# 遍历所有行程
for trip in onebike_datetimes:
    # 检查行程是否在中午前开始
    if trip['start'].hour < 12:
        # 增加上午计数
        trip_counts["AM"] += 1
    else:
        # 增加下午计数
        trip_counts['PM'] += 1

你尝试的列表推导式代码报错，是因为trip_counts["AM"] +=1是赋值语句，不是一个能返回值的表达式——列表推导式的第一个位置必须是可以生成元素的表达式，而不是用来执行修改操作的语句。而且这种修改外部变量的“副作用”，也违背了列表推导式的设计初衷。

正确的两种写法 #

写法1：用生成器表达式+sum()统计（更Pythonic）

这种方式直接计算出AM和PM的数量，不需要先初始化字典再修改，代码更简洁：

# 统计上午行程数：符合条件就加1，用sum累加
am_count = sum(1 for trip in onebike_datetimes if trip['start'].hour < 12)
# 下午行程数=总行程数-上午数
pm_count = len(onebike_datetimes) - am_count

# 生成结果字典
trip_counts = {'AM': am_count, 'PM': pm_count}

写法2：如果一定要保留原字典的初始化逻辑（不推荐用推导式，循环更清晰）

如果你坚持要基于已有的trip_counts字典更新，其实原有的for循环已经很清晰了，可读性比强行用推导式更好。如果非要用推导式的思路（不推荐，因为不直观），可以用生成器表达式配合any()来触发遍历执行，但这属于技巧性写法，不适合日常代码：

trip_counts = {'AM': 0, 'PM': 0}
# 用any()遍历生成器，触发赋值操作（不推荐，可读性差）
any(trip_counts['AM'] += 1 if trip['start'].hour < 12 else trip_counts['PM'] += 1 for trip in onebike_datetimes)

总结 #

列表推导式适合用来快速生成列表，比如[trip['start'].hour for trip in onebike_datetimes]这种场景。而涉及到修改外部变量、执行副作用操作的逻辑，用普通for循环会更清晰易懂，也符合Python的“可读性优先”原则。

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