如何在Pandas中用apply()生成含表格与单值的新列？

你遇到的问题大概率是参数传递格式不匹配或者apply()对多返回值的处理方式没搞对，和函数本身兼容apply()关系不大——毕竟你单独调用函数是正常的。

常见问题点及解决步骤 #

1. 函数参数是否适配Series类型？ #

table1的每一行是pd.Series类型，如果你的calculate_table()函数是按列名提取行数据（比如row['col_name']），那没问题；但如果函数定义时要求的是单个数值参数，而非整行数据，肯定会报错。

• 检查函数定义，确保它能接收整行Series：

  def calculate_table(row):
      # 从行中提取所需字段
      col1_val = row['col1']
      col2_val = row['col2']
      # 生成子DataFrame、p和x的最终值
      sub_table = pd.DataFrame(...)
      p_last = ...
      x_last = ...
      return sub_table, p_last, x_last
  

2. apply()未正确处理多返回值 #

默认情况下，df.apply()会把函数返回的多个值打包成元组塞进一列，不会自动拆分成多列。你需要通过result_type='expand'参数让它自动展开：

• 正确调用apply的方式：

  # 让返回的三个值直接对应新增的三列
  table1[['table', 'p', 'x']] = table1.apply(calculate_table, axis=1, result_type='expand')
  

3. 子表数据的正确访问方式 #

你期望的table1['table']['w'][<index>]可能会因为索引匹配问题失效，正确的访问姿势应该是：

# 访问第i行子DataFrame的'w'列
table1.loc[i, 'table']['w']
# 或者用iloc按位置取
table1['table'].iloc[i]['w']

快速排查步骤 #

1. 先单独测试：calculate_table(table1.iloc[0])，确认返回的是（子DataFrame，p值，x值）的结构。
2. 如果单独调用正常，给apply加上result_type='expand'参数，再执行。
3. 若仍报错，把具体错误信息贴出来——比如是参数类型不兼容还是返回值结构不对，能更快定位问题。

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