我一直在尝试转换一些数据,每个组中的第一个值是组标识符:
data = [(a, 1, 2), (a, 3, 4), (b, 5, 6)]
进入
data = [[(1, 2), (3, 4)], [(5, 6)]]
最简单的方法是什么?提前致谢
最佳答案
如果data
列表是按分组键排序的,你可以使用itertools.groupby
:
In [1]: from itertools import groupby
In [2]: from operator import itemgetter
In [3]: data = [(a, 1, 2), (a, 3, 4), (b, 5, 6)]
In [4]: [[l[1:] for l in group] for _, group in groupby(data, key=itemgetter(0))]
Out[4]: [[(1, 2), (3, 4)], [(5, 6)]]
关于python - 列表基于 Python 中的值重新分组,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41445702/