Pandas DataFrame列表类型列按条件追加元素报错如何高效解决

报错原因 #

你原先的写法触发类型错误，核心是df['list_of_purchases'] + ['Null']的逻辑不符合Pandas的广播规则：Pandas会尝试把长度为1的列表['Null']和长度为N的列表类型Series做广播匹配，而列表类型的Series不支持这种广播操作，因此报错。

高效实现方案 #

方案1：map+Series加法（性能最优，优先推荐） #

先通过map生成和原数据集行数一致的待追加元素Series，再直接和原列表列相加，全程为Pandas优化过的矢量化操作，性能远高于手动for循环和逐行apply：

# 生成每行待追加的元素列表
append_series = df['b'].map({0: ['Null'], 1: ['Success']})
# 直接拼接两个列表类型Series
df['final_list'] = df['list_of_purchases'] + append_series

方案2：apply逐行判断（写法更简洁，适合中小数据集） #

如果你的数据量在10万行以下，用apply的写法更直观易读：

df['final_list'] = df.apply(
    lambda row: row['list_of_purchases'] + ['Success'] if row['b'] == 1 else row['list_of_purchases'] + ['Null'],
    axis=1
)

注意事项 #

如果list_of_purchases列存在非列表格式的空值，需要先做清洗再执行上述操作，避免拼接报错：

# 把非列表的空值统一替换为空列表
df['list_of_purchases'] = df['list_of_purchases'].apply(lambda x: x if isinstance(x, list) else [])

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