按地名分组在python中

Question

我想按列表中的列表进行分组。下面的列表给出：

{'San Antonio,TX': [[-50,98],2, 3]} 

更精确地在输出内部列表的第一个元素必须要么2个第一元素：

[[[-50,98],1,2,'San Antonio,TX'],[[-52,99],3,4,'San Antonio,TX']] 

我期待的输出在下面给出。输出中内部列表的第二个元素是输入内部列表中第二个元素的平均值。输出中内部列表的第三个元素是输入内部列表中第三个元素的平均值。

这实际上是一个问题的小版本。我拥有的名单的初始名单将有1000或更多的名单。

感谢

Answer 1

test.py内容

from itertools import groupby 
from operator import itemgetter 


def aggregate(data): 
    result = [] 
    data = sorted(data, key=itemgetter(3)) 
    for k, g in groupby(data, itemgetter(3)): 
     g = list(g) 
     n = len(g) # Cast to float if you want average to be float in python 2 
     x1 = g[0][0] 
     x2 = sum(x[1] for x in g) 
     x3 = sum(x[2] for x in g) 
     d = {k: [x1, x2/n, x3/n]} # This is what you want for each group 
     result.append(d) 

    return result 


def main(): 
    data = [[[-50,98], 1, 2, 'San Antonio,TX'], 
      [[-52,99], 3, 4, 'San Antonio,TX']] 
    result = aggregate(data) 
    print(result) 

if __name__ == '__main__': 
    main() 

结果：

In [12]: run test.py 
[{'San Antonio,TX': [[-50, 98], 2, 3]}]