python中等于列表的某一个值为真,python – 获取值等于特定值的列表中的所有元素...

这应该工作：

lst = [[3, 4.6575, 7.3725],

[3, 3.91, 5.694],

[2, 3.986666666666667, 6.6433333333333335],

[1, 3.9542857142857137, 5.674285714285714]]

# group the values in a dictionary

import collections

d = collections.defaultdict(list)

for item in lst:

d[item[0]].append(item)

# find sum of values

for key, value in d.items():

print [key] + map(sum, zip(*value)[1:])

或者,使用itertools.groupby更清洁一点：

import itertools

groups = itertools.groupby(lst, lambda i: i[0])

for key, value in groups:

print [key] + map(sum, zip(*value)[1:])

两种情况下的输出：

[1, 3.9542857142857137, 5.674285714285714]

[2, 3.986666666666667, 6.6433333333333335]

[3, 8.567499999999999, 13.0665]

如果你想计算均值而不是总和,只需定义你自己的均值函数并传递一个而不是sum函数来映射：

mean = lambda x: sum(x) / float(len(x))

map(mean, zip...)