在列表Python中对相同的元素进行分组

在列表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

一个公正的笔记 - 这只有在列表按您想要分组的项目排序时才有效。 '[(1,'a'),(3,'b'),(2,'a')]'不会给出[[1,2],[3]]'而是[[1], [2],[3]]'! – 2015-02-12 04:25:29