DataFrame为空时pandas apply方法未按预期工作的原因与解决

问题原因 #

当原DataFrame为空时，apply(axis=1)不会执行你定义的func函数——因为没有行可供遍历。Pandas无法从func的返回值推断新的列结构，所以直接沿用原DataFrame的列名返回空DataFrame。

解决方案 #

1. 手动校验并重置空DataFrame的列 #

如果一定要用apply，可以在生成结果后判断是否为空，手动指定目标列：

import pandas

def func(row): 
    return pandas.Series({'d': row['a'], 'e': row['b'] + row['c']})

df = pandas.DataFrame(columns=['a', 'b', 'c'])
df_new = df.apply(func, axis=1)

# 处理空数据情况
if df_new.empty:
    df_new = pandas.DataFrame(columns=['d', 'e'])
df_new = df_new.reset_index(drop=True)

print(df_new)

输出结果：

Empty DataFrame
Columns: [d, e]
Index: []

2. 用向量化操作替代apply（推荐） #

你的逻辑可以完全不用逐行遍历的apply，直接用Pandas的向量化运算生成新DataFrame，这种方式既高效，又能自动处理空数据的列结构问题：

import pandas

df = pandas.DataFrame(columns=['a', 'b', 'c'])
df_new = pandas.DataFrame({
    'd': df['a'],
    'e': df['b'] + df['c']
}).reset_index(drop=True)

print(df_new)

这种方法直接基于原列生成新列，不管原DataFrame是否为空，都会生成指定的d、e列。

关于apply参数的说明 #

Pandas的apply方法本身没有可以直接指定返回列名的参数，无法仅通过添加参数解决这个问题，必须通过额外的逻辑处理或者换用更合适的写法。

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