python – 获取值等于特定值的列表中的所有元素

[[3, 4.6575, 7.3725], 
[3, 3.91, 5.694],
[2, 3.986666666666667, 6.6433333333333335],
[1, 3.9542857142857137, 5.674285714285714],....]

我想总和(实际上取平均值…但它是一个细节)所有行的值在一起,其中第一个元素的值相等.这意味着在上面的例子中,前两行将被加在一起.

[[3, 8.5675, 13.0665],
[2, 3.986666666666667, 6.6433333333333335],
[1, 3.9542857142857137, 5.674285714285714],....]

这意味着第一个值应该是唯一的.

我想通过找到第一个值等于例如1的所有“行”并将它们加在一起来做到这一点.我现在的问题是,如何找到第一个值等于某个值的所有行.

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...)