布尔比较中字符串引用问题：多条件循环赋值KeyError报错排查

我来帮你捋清楚这个问题——你遇到的KeyError完全是因为循环里条件写法的问题，和三个变量本身没关系！

错误的根源 #

你在循环里把条件写成了字符串拼接后直接比较，比如：

('results.Pre_BF_' + str(i) + str(k) == 1)

这行代码的实际作用是判断字符串"results.Pre_BF_2014_Flag"是否等于整数1，结果只会返回一个单个的布尔值False（因为字符串和整数永远不相等）。而Pandas的.loc索引需要的是和DataFrame行数匹配的布尔Series（每一行对应一个判断结果），用单个布尔值去索引自然就触发了cannot use a single bool to index into setitem的错误。

反观你硬编码的代码：

results.loc[(results.Pre_BF_2014_Flag == 1) & (results.BF_2014_Flag == 0) & (results.XMAS_2014_Flag == 0), 'Combo_2014_Pre_BF_Only'] = 1

这里results.Pre_BF_2014_Flag == 1是直接访问DataFrame的列，返回的是和DataFrame行数一致的布尔Series，所以能正常工作。

解决方案 #

有两种简单的方式修正循环里的条件写法，核心都是正确动态获取DataFrame的列，生成布尔Series：

1. 使用f-string直接引用列名

利用Python的f-string语法构造列名，直接访问DataFrame的列：

import pandas as pd
import numpy as np

# 生成测试数据集
results = pd.DataFrame(np.random.randint(0,2,size=(20, 3)), columns=['Pre_BF_2014_Flag', 'BF_2014_Flag', 'XMAS_2014_Flag' ])

years= [2014, 2015, 2016, 2017, 2018]
var = ['_Flag']
for i in years:
    for k in var:
        # 用f-string构造列名，生成匹配行数的布尔掩码
        mask = (results[f'Pre_BF_{i}{k}'] == 1) & \
               (results[f'BF_{i}{k}'] == 0) & \
               (results[f'XMAS_{i}{k}'] == 0)
        # 注意目标列名也应该对应年份，避免循环覆盖同一列
        results.loc[mask, f'Combo_{i}_Pre_BF_Only'] = 1

2. 使用getattr()方法获取列

如果习惯用属性访问的方式，可以用getattr动态获取DataFrame的列：

import pandas as pd
import numpy as np

results = pd.DataFrame(np.random.randint(0,2,size=(20, 3)), columns=['Pre_BF_2014_Flag', 'BF_2014_Flag', 'XMAS_2014_Flag' ])

years= [2014, 2015, 2016, 2017, 2018]
var = ['_Flag']
for i in years:
    for k in var:
        mask = (getattr(results, f'Pre_BF_{i}{k}') == 1) & \
               (getattr(results, f'BF_{i}{k}') == 0) & \
               (getattr(results, f'XMAS_{i}{k}') == 0)
        results.loc[mask, f'Combo_{i}_Pre_BF_Only'] = 1

另外补充一句：你原来的目标列名Combo_2014_Pre_BF_Only是固定的2014，循环里改成对应年份的列名（比如f'Combo_{i}_Pre_BF_Only'），才能避免每次循环都覆盖同一列的数据。

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