我有一个列表
a = [[1,2,3],[4,5,6],[7,8,9]]
现在我想找到这些内部列表的平均值,以便
a = [(1+4+7)/3,(2+5+8)/3,(3+6+9)/3]
'a' 最终不应该是一个嵌套列表。请提供一般情况下的答案。
a = [[1,2,3],[4,5,6],[7,8,9]]
现在我想找到这些内部列表的平均值,以便
a = [(1+4+7)/3,(2+5+8)/3,(3+6+9)/3]
'a' 最终不应该是一个嵌套列表。请提供一般情况下的答案。
a = [sum(x)/len(x) for x in zip(*a)]
# a is now [4, 5, 6] for your example
在Python 2.x中,如果你不想使用整数除法,请在sum(x)/len(x)
前加上1.0*
如果您已经安装了numpy:
>>> import numpy as np
>>> a = [[1,2,3],[4,5,6],[7,8,9]]
>>> arr = np.array(a)
>>> arr
array([[1, 2, 3],
[4, 5, 6],
[7, 8, 9]])
>>> np.mean(arr)
5.0
>>> np.mean(arr,axis=0)
array([ 4., 5., 6.])
>>> np.mean(arr,axis=1)
array([ 2., 5., 8.])
>>> import itertools
>>> [sum(x)/len(x) for x in itertools.izip(*a)]
[4, 5, 6]