Python计算标准差报错：'float' object is not iterable及range问题排查

我尝试编写程序计算给定数值列表的平均值和标准差，平均值功能正常，但计算标准差（sd）时出现'float' object is not iterable错误。后来尝试用range函数遍历列表，又出现'list' object cannot be interpreted as an integer错误。

原代码如下：

from math import sqrt
def mean(xs):
    return sum(xs)/len(xs)

def sd(xs):
    for x in xs:
        a=sqrt(sum(x-mean(xs)**2)/(len(xs)-1))
    return a

def computeMeasure(xs, unit):
    return (mean(xs), sd(xs), unit)


xs = [11,12,13,14,11,11,12,10,15]
print(f'{mean(xs):6.3f}, {sd(xs):5.3f}')

错误原因分析 #

• 第一个错误：sd函数里，sum(x-mean(xs)**2)中的x是列表里的单个浮点数，而sum()要求传入可迭代对象（比如列表），直接传单个数值会触发迭代错误。同时这个for循环完全多余，每次循环都重复计算整个标准差，没有实际作用。
• 第二个错误：range(xs)中，range()需要整数参数（比如列表长度），直接传入列表会导致类型不匹配错误。

修正后的代码 #

from math import sqrt

def mean(xs):
    return sum(xs) / len(xs)

def sd(xs):
    avg = mean(xs)
    # 计算每个元素与均值的差的平方，直接求和
    squared_diff_sum = sum((x - avg) ** 2 for x in xs)
    # 样本标准差用n-1做分母，总体标准差改用len(xs)即可
    return sqrt(squared_diff_sum / (len(xs) - 1))

def computeMeasure(xs, unit):
    return (mean(xs), sd(xs), unit)


xs = [11,12,13,14,11,11,12,10,15]
print(f'{mean(xs):6.3f}, {sd(xs):5.3f}')

说明 #

1. 提前计算一次平均值，避免循环中重复调用mean(xs)，提升运行效率。
2. 用生成器表达式直接计算并求和所有元素与均值差的平方，写法简洁高效。
3. 样本标准差公式中分母用len(xs)-1，如果需要计算总体标准差，将分母改为len(xs)即可。

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