pandas使用apply为DataFrame创建多列时不同行数量输出结果不一致问题

核心原因 #

1. pandas的apply在沿axis=1执行时，会默认采样前2行的返回结果推断整体返回结构：2行场景下所有行返回结构一致，推断正确，赋值正常；20行场景下前2行推断完结构后，后续行出现了和前2行不一致的返回结构，导致整体返回结构不符合预期，赋值后数据错位。
2. 你收到的警告翻译为：弃用警告：未来版本中空Series的默认dtype将从float64改为object，请显式指定dtype以消除此警告。该警告直接说明你的自定义函数在部分行执行时返回了空Series，这就是返回结构不一致的核心诱因。

修复步骤 #

• 首先统一自定义函数的返回结构：无论计算逻辑是否命中异常分支，都返回固定长度的结果。比如你需要返回3个值对应3个新列，即使某行计算失败，也要返回pd.Series([np.nan, np.nan, np.nan], dtype='float64')这类固定长度的对象，禁止返回空Series或者长度与预期不符的结果。
• 给apply调用添加result_type='expand'参数，强制将每行返回结果展开为多列，避免pandas自行推断结构出错，修改后的调用示例如下：

df[new_columns] = df.apply(function_name, axis=1, result_type='expand')

• 调试阶段可以先单独接收apply的返回结果，确认结构符合预期后再赋值，避免直接赋值打乱原DataFrame数据：

# 先接收返回结果检查结构
apply_res = df.apply(function_name, axis=1, result_type='expand')
print(apply_res.shape, apply_res.dtypes)
# 确认列数、dtype和预期一致后再赋值
df[new_columns] = apply_res

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