重组我的以下数据的 pythonic 方法是什么?
我有一个数据
data = [
['a','b',1], ['a','b',2], ['a','b',3],
['a','c',3], ['a','c',4],
['f','g',2], ['f','g',5], ['f','g',9]
]
我想将其重新排列为以下格式:
data = [
['a', 'b', 1, 2, 3],
['a', 'c', 3, 4],
['f', 'g', 2, 5, 9]
]
所以基本上每个内部列表中的前两个元素是区分不同项目的方式,后面的数字是数据。我只想为包含所有数据的每个项目留一行。
最佳答案
import collections
keyed = collections.defaultdict(list) # (a,b): [1,2,3]
for k1,k2,val in data:
keyed[(k1,k2)].append(val)
[list(keys) + vals for keys,vals in sorted(keyed.items())]
关于列表重新格式化中的 Python 列表,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/36861715/