Python矩阵元素均值计算问题求助：仅用for循环与列表推导式

先明确你代码里的核心问题：

• 语法错误：range(v)中的v是列表，而range()要求传入整数参数，这就是触发TypeError的直接原因。
• 逻辑错误：循环刚遍历到第一个向量就执行return，完全没处理后续向量；且当前逻辑是计算单个向量的均值，并非你需要的逐元素（列）均值。

要实现目标，本质是对矩阵的每一列求平均：先将矩阵转置（把列转为行），再对每一行（原矩阵的列）计算均值。以下是两种符合你要求的实现方式：

方式一：列表推导式实现 #

def vector_mean(vectors):
    """Computes the element-wise average"""
    # zip(*vectors)将所有向量的对应元素打包成元组（实现矩阵转置）
    # 对每个元组求和后，除以向量的总个数得到该位置的均值
    return [sum(col) / len(vectors) for col in zip(*vectors)]

assert vector_mean([[1, 2], [3, 4], [5, 6]]) == [3, 4]
print(vector_mean([[1, 2], [3, 4], [5, 6]]))  # 输出 [3.0, 4.0]

方式二：纯for循环实现 #

如果不想使用zip，仅靠for循环完成：

def vector_mean(vectors):
    """Computes the element-wise average"""
    # 假设所有向量长度一致（矩阵的基本要求）
    elem_num = len(vectors[0])
    result = []
    # 遍历每个元素位置
    for i in range(elem_num):
        total = 0
        # 累加所有向量对应位置的元素
        for v in vectors:
            total += v[i]
        # 计算均值并加入结果列表
        result.append(total / len(vectors))
    return result

assert vector_mean([[1, 2], [3, 4], [5, 6]]) == [3, 4]
print(vector_mean([[1, 2], [3, 4], [5, 6]]))  # 输出 [3.0, 4.0]

补充说明 #

两种实现均默认输入为规整矩阵（所有向量长度相同），若需处理长度不一致的情况，可额外添加参数校验逻辑。

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