在列表Python中对相同的元素进行分组
问题描述:
如果第一个元素使用下面的代码具有相同的第二个元素,我已经设法将第一个元素分组到列表中。有没有更好的方式来做到这一点。在列表Python中对相同的元素进行分组
注意:总是有两个相同的第二个元素。
lst = [
[1, '200309060143'],
[2, '200309060143'],
[3, '200309060143'],
[4, '200309060143'],
[5, '200309060143'],
[6, '200309060143'],
[7, '200309060143'],
[8, '200309060143'],
[1, '200309060144'],
[2, '200309060144'],
[3, '200309060144'],
[1, '200309060145'],
[2, '200309060145'],
[1, '200401100047'],
[2, '200401100047'],
[3, '200401100047']
]
mega_lst = []
temp_lst = []
for i in range(len(lst)):
if i == (len(lst)-1):
break;
else:
if lst[i][1] == lst[i + 1][1]:
temp_lst.append(lst[i][0])
if i == (len(lst)-2):
temp_lst.append(lst[len(lst)-1][0])
mega_lst.append(temp_lst)
else:
temp_lst.append(lst[i][0])
mega_lst.append(temp_lst)
temp_lst = []
print mega_lst
我的代码结果:[[1,2,3,4,5,6,7,8],[1,2,3],[1,2],[1,2,3 ]]
预期结果:[[1,2,3,4,5,6,7,8],[1,2,3],[1,2],[1,2,3]]
答
>>> list(list(x[1]) for x in itertools.groupby(lst, operator.itemgetter(1)))
[[[1, '200309060143'], [2, '200309060143'], [3, '200309060143'], [4, '200309060143'], [5, '200309060143'], [6, '200309060143'], [7, '200309060143'], [8, '200309060143']], [[1, '200309060144'], [2, '200309060144'], [3, '200309060144']], [[1, '200309060145'], [2, '200309060145']], [[1, '200401100047'], [2, '200401100047'], [3, '200401100047']]]
编辑:
而如果下一步就是给你的麻烦:
>>> list([y[0] for y in x[1]] for x in itertools.groupby(lst, operator.itemgetter(1)))
[[1, 2, 3, 4, 5, 6, 7, 8], [1, 2, 3], [1, 2], [1, 2, 3]]
一个公正的笔记 - 这只有在列表按您想要分组的项目排序时才有效。 '[(1,'a'),(3,'b'),(2,'a')]'不会给出[[1,2],[3]]'而是[[1], [2],[3]]'! – 2015-02-12 04:25:29