如何编写函数向Pandas DataFrame传递运算符以实现指定列的统计计算

首先先确认你的DataFrame定义：

import pandas as pd
tmp = pd.DataFrame([(2,2),(0,4),(1,3)],columns= ['A','B'])

要实现你想要的f(operator, column)函数，核心思路是通过字符串动态获取Pandas Series的对应方法，这里可以用Python内置的getattr()函数来实现，具体代码如下：

def f(operator, column):
    # 第一步：获取指定列的Series对象
    target_series = tmp[column]
    # 第二步：根据operator字符串获取对应的方法（比如'mean'对应series.mean）
    calc_method = getattr(target_series, operator)
    # 第三步：调用方法并返回计算结果
    return calc_method()

测试一下调用效果：

print(f('mean','A'))  # 输出 1.0，和tmp['A'].mean()结果一致
print(f('sum','B'))   # 输出 9，对应tmp['B'].sum()的结果

可选：添加异常处理增强鲁棒性 #

如果担心传入不存在的列名或不支持的操作符，可以加上异常处理，让函数更友好：

def f(operator, column):
    try:
        target_series = tmp[column]
        calc_method = getattr(target_series, operator)
        return calc_method()
    except KeyError:
        return f"错误：DataFrame中不存在列名 '{column}'"
    except AttributeError:
        return f"错误：Pandas Series不支持 '{operator}' 这个操作"

这个方法的逻辑很直观：getattr()可以根据字符串名称，获取对象的属性或可调用方法，完美匹配你需要的“用字符串指定操作”的需求。

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