python - 平均列表python列

Question

我有一个列表列表:类似:

data = [[240, 240, 239],
        [250, 249, 237], 
        [242, 239, 237],
        [240, 234, 233]]

我想平均一下

[average_column_1, average_column_2, average_column_3]

我的代码不是很优雅。这是遍历列表的幼稚方式，将总和保存在单独的容器中，然后除以元素的数量。

我认为有一种 Pythonic 方法可以做到这一点。有什么建议么？谢谢

Answer 1

纯 Python:

from __future__ import division
def mean(a):
    return sum(a) / len(a)
a = [[240, 240, 239],
     [250, 249, 237], 
     [242, 239, 237],
     [240, 234, 233]]
print map(mean, zip(*a))

打印

[243.0, 240.5, 236.5]

NumPy:

a = numpy.array([[240, 240, 239],
                 [250, 249, 237], 
                 [242, 239, 237],
                 [240, 234, 233]])
print numpy.mean(a, axis=0)

Python 3:

from statistics import mean
a = [[240, 240, 239],
     [250, 249, 237], 
     [242, 239, 237],
     [240, 234, 233]]
print(*map(mean, zip(*a)))