Python的groupby

今天在一个群里看到一个问题



可能是Java用得久了，第一反应就是把使用一个HashMap ,然后便历属性，便可以做到了。用python实现一下：

# -*- coding: utf-8 -*-
import json

# Create your tests here.
j = json.loads('[{"id":1,"name":"a"},{"id":1,"name":"b"},{"id":1,"age":32},{"id":2,"age":32},{"id":2,"name":"22"}]')
d = {}

def biz(a):
try:
v = d[a['id']]
for kk, vv in a.iteritems():
if hasattr(v, kk):
if isinstance(vv, unicode):
v[kk] = vv
else:
v[kk] = vv if vv > v[kk] else v[kk]
else:
v[kk] = vv
except:
d[a['id']] = a

map(biz, j)
print(d)

写得不太简洁，但是可以实现需求，想想那些一行python代码，就找了一下python的相关资料，就发现了：https://docs.python.org/2/library/itertools.html#itertools.groupby

这个函数功能类似于SQL的分组。使用groupby前，首先需要使用相同的keyfunc对iterable进行排序，比如调用内建的sorted函数。然后，groupby返回迭代器，每次迭代的元素是元组(key值, iterable中具有相同key值的元素的集合的子迭代器)。

简单的说，就是对于可迭代的对象，可以 以指定的元素进行分组，比如我现在所遇到的情况，就可以使用 元素的 id进行分组。分组后数据是这样的:

[{k1:[{},{}],k2:[{},{}]}]

上面的写法是手动分组，现在使用group by 函数帮我们处理，就可以简化为：

# -*- coding: utf-8 -*-
import json
from itertools import groupby
j = json.loads('[{"id":1,"name":"a"},{"id":1,"name":"b"},{"id":1,"age":32},{"id":2,"age":32},{"id":2,"name":"22"}]')
result = []
def merge(a, b):
for k, v in a.iteritems():
b[k] = v if k not in b or isinstance(b[k], unicode) else b[k] if b[k] > v else v
return b

for k, group in groupby(j, lambda x: x['id']):
result.append(reduce(lambda a, b: merge(a, b), group))
print(result)