Python编写返回均值、中位数、截尾均值的函数问题排查

你的代码存在几个关键问题，导致无法得到预期结果：

• 参数定义完全错误：需求要求函数接收x和字符串类型的type参数（指定计算类型），但你把第二个参数写成了median，这会让函数把传入的第二个值当成函数来调用，完全偏离需求。
• 缺少多类型分支逻辑：当前函数只写了中位数的调用逻辑，完全没处理均值、截尾均值的计算场景。
• 中位数调用方式错误：直接写median(x_vec)会触发NameError，因为你没有定义这个函数，应该使用numpy提供的np.median()方法。
• 缺失截尾均值依赖：需求里要用到scipy.stats.trim_mean，但你的代码没有导入scipy模块。

修正后的完整代码 #

import numpy as np
from scipy.stats import trim_mean

x_vec = np.array([1,2,3,4,5])

def f(x, type):
    if type == "mean":
        return np.mean(x)
    elif type == "median":
        return np.median(x)
    elif type == "trimmed":
        return trim_mean(x, 0.1)
    else:
        raise ValueError("type参数仅支持'mean'、'median'、'trimmed'三种取值")

# 测试验证
print(f(x_vec, "mean"))      # 输出 3.0
print(f(x_vec, "median"))    # 输出 3.0
print(f(x_vec, "trimmed"))   # 输出 3.0

修正说明 #

1. 调整函数参数为(x, type)，完全匹配需求的参数定义
2. 通过if-elif分支实现不同统计量的计算逻辑
3. 正确调用numpy和scipy提供的统计方法
4. 增加错误处理，避免传入无效的type参数导致逻辑混乱

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