Pandas为匹配条件的记录创建DataFrame列赋值遇KeyError问题

问题原因 #

你用了链式索引（dataframe.loc[...]["new_column"]），这会触发两个核心问题：

1. 因为new_column尚未存在，链式索引的第二步直接抛出KeyError
2. 就算列已存在，链式索引大概率返回原DataFrame的副本，赋值操作只会修改副本，原DataFrame不会产生任何变化

正确写法 #

方法1：用loc一次性完成条件筛选与列赋值 #

直接通过loc同时指定行条件和目标列，Pandas会自动创建不存在的列，且修改直接作用于原DataFrame：

import pandas as pd

criteria = ["XX", "XY", "XYY"]
dataframe.loc[dataframe["criteria_column"].isin(criteria), "new_column"] = 1000

# 查看结果
dataframe["new_column"]

这是Pandas官方推荐的操作方式，避免了副本与视图的混淆问题。

方法2：先初始化列再赋值（可选） #

如果需要给不符合条件的行设置默认值（比如NaN或0），可以先创建列再用loc赋值：

import pandas as pd

# 先初始化列，默认值设为NaN
dataframe["new_column"] = pd.NA
# 给符合条件的行赋值
dataframe.loc[dataframe["criteria_column"].isin(criteria), "new_column"] = 1000

原代码失效的本质 #

原代码的dataframe.loc[...]["new_column"]相当于先提取符合条件的行生成一个临时对象，再试图给这个临时对象的new_column赋值——但这个临时对象里根本没有new_column，所以直接触发KeyError。就算不报错，修改的也只是临时对象，原DataFrame不会有任何变化。

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