Pandas自定义函数结果无法作为新列添加到DataFrame问题排查

新列赋值后不显示，本质是以下三类问题导致的，你贴的代码存在全部三类问题的触发可能：

• 链式索引引发的视图操作：你当前持有的df是其他DataFrame切片生成的临时视图，对视图直接赋值新列不会同步到你最终打印的原始对象，pandas通常会抛出SettingWithCopyWarning提示，但多数场景下开发者会忽略控制台的警告信息。
• 生成器输出长度和DataFrame行数不匹配：你写的比对函数只处理了int和list两种类型，只要某行数据属于其他类型（比如字符串、浮点数、空值），循环走到对应分支就不会执行yield，最终转出来的result列表长度比df行数短，旧版本pandas遇到长度不匹配的赋值会静默失败，不会添加新列也不会抛出明确报错。
• 作用域问题：如果上述逻辑写在自定义函数内部，你修改的是函数局部作用域的df变量，函数外全局作用域的df没有被改动，打印时自然看不到新列。

按以下步骤调整代码即可解决：

1. 补全比对函数的分支逻辑，保证每一行数据都能返回一个比对结果，同时替换不安全的eval为安全的类型解析方法：

from ast import literal_eval

def custom_compare_eq(series, other):
    length = len(series.values)
    for i in range(length):
        # 加异常捕获，处理值无法解析的场景
        try:
            r1 = literal_eval(str(series.values[i]))
            r2 = literal_eval(str(other.values[i]))
        except:
            yield False
            continue
        
        if type(r1) != type(r2):
            yield False
        else:
            if isinstance(r1, int):
                yield r1 == r2
            elif isinstance(r1, list):
                yield set(r1) == set(r2)
            else:
                # 补全其余类型的判断逻辑，保证每行必有返回
                yield r1 == r2

2. 赋值前增加长度校验，同时强制使用DataFrame副本操作，切断视图关联：

print('store the result output')
result = list(custom_compare_eq(df.a_series, df.b_series))
# 校验长度，不匹配直接抛错，避免静默失败
assert len(result) == len(df), f"比对结果长度{len(result)}与DataFrame行数{len(df)}不一致"
print('add new match column to df')
# 强制生成副本，避免视图赋值失效
df = df.copy()
df['match'] = result
print(df)

3. 如果逻辑封装在函数内，必须将修改后的DataFrame作为返回值传出，调用时重新接收返回值：

def process_df(input_df):
    df = input_df.copy()
    result = list(custom_compare_eq(df.a_series, df.b_series))
    df['match'] = result
    # 返回修改后的对象
    return df

# 调用时重新赋值
df = process_df(df)

注意：eval会直接执行传入字符串的任意代码，存在严重安全风险，处理结构化数据的类型解析时，优先使用ast.literal_eval，该方法只会解析Python原生数据类型，不会执行恶意代码。

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