为何Python字典推导式报错而for循环正常？异常解析

调用示例函数func时，启用字典推导式（通过comprehension参数切换）会抛出异常，而使用普通for循环时可正常运行。需要解释该异常的含义，以及为何字典推导式会出现cycle变量未定义的问题。

循环与推导式实现相同功能的示例函数 #

import pandas as pd

def func(
    x_dict,
    keys_list,
    start_cycle,
    end_cycle,
    comprehension=True
):
    x_test_dicts = {}
    for cycle in range(start_cycle, end_cycle + 1):
        print(f"cycle = {cycle}")
        if comprehension:
            # 使用字典推导式填充字典
            x_test_dict = {
                f"{key}_input":
                x_dict[key].query('cycle == @cycle').values
                for key in keys_list
            }
        else:
            # 使用普通for循环填充字典
            x_test_dict = {}
            for key in keys_list:
                x_test_dict[f"{key}_input"] = \
                    x_dict[key].query('cycle == @cycle').values
        x_test_dicts[cycle] = x_test_dict
    return x_test_dicts

测试数据生成 #

import pandas as pd
import numpy as np

# 创建1到1000的ID数组
ids = np.arange(1, 1001)

# 通过ID除以100计算cycle值
cycles = ids // 100

# 生成其余列的随机整数值（范围0-100）
col1_int = np.random.randint(0, 101, 1000)
col2_int = np.random.randint(0, 101, 1000)
col3_int = np.random.randint(0, 101, 1000)

# 创建包含整数值的DataFrame
df = pd.DataFrame({
    "ID": ids,
    "cycle": cycles,
    "col1": col1_int,
    "col2": col2_int,
    "col3": col3_int
})

df.head()  # 显示DataFrame的前几行

函数测试用例运行 #

import pandas as pd

df = df.set_index(['ID', 'cycle'])  # 使用多重索引

x_dict = {'Auxin': df}  # 创建包含DataFrame的字典
keys_list = ['Auxin']  # 定义要处理的键列表

# 定义func内部循环的范围
start_cycle = 6
end_cycle = 29

# 不使用字典推导式，运行成功
comprehension = False
result = func(
    x_dict,
    keys_list,
    start_cycle,
    end_cycle,
    comprehension=comprehension
)
print("不使用字典推导式时运行成功！")

# 使用字典推导式，运行失败
comprehension = True
result = func(
    x_dict,
    keys_list,
    start_cycle,
    end_cycle,
    comprehension=comprehension
)
print("使用字典推导式时运行失败！")

报错信息 #

UndefinedVariableError: local variable 'cycle' is not defined

异常含义 #

UndefinedVariableError表示pandas的query方法在当前作用域中找不到名为cycle的变量，无法完成你指定的查询逻辑。

字典推导式导致变量未定义的原因 #

Python的字典推导式会创建独立的嵌套局部作用域，而普通for循环则与外层代码共享同一作用域：

1. 普通for循环分支中，query里的@cycle引用的是外层for cycle in range(...)中的cycle变量，这个变量处于函数的局部作用域，query可以正常识别并调用。
2. 字典推导式内部会形成一个新的隔离作用域，外层的cycle变量不会被自动导入到这个作用域中。当query尝试通过@cycle引用变量时，只能检索推导式内部的作用域，而这里并没有定义cycle，因此抛出未定义的异常。

简单来说，字典推导式的作用域隔离特性，导致query无法访问外层循环的cycle变量，而普通for循环不存在这个作用域隔离问题。

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