Python_grouping多维列表

Question

我有一个多维列表示例：

    example_list=[
        ["a","b","c", 2,4,5,7],
        ["e","f","g",0,0,1,5],
        ["e","f","g", 1,4,5,7],
        ["a","b","c", 3,2,5,7]
    ]

怎么可能把他们分成这样的小组：

out_list=[
         [["a","b","c", 2,4,5,7],
           ["a","b","c",3,2,5,7]
         ],
         [["e","f","g", 0,0,1,5],
          ["e","f","g", 1,4,5,7]
         ]
 ]

我试过这样做：

example_list=[["a","b","c", 2,4,5,7],["e","f","g", 0,0,1,5],["e","f","g",1,4,5,7],["a","b","c", 3,2,5,7]]
unique=[]
index=0
for i in range(len(example_list)):
    newlist=[]
    if example_list[i][:3]==example_list[index][:3]:
        newlist.append(example_list[i])
    index=index+1
    unique.append(newlist)            
print unique

我的结果是：

[['a'，'b'，'c'，2，4，5，7]，['e'，'f'，'g',0，0，1，5]，[e‘，'f'，'g'，1，4，5，7]，['a'，'b'，'c'，3，2，5,7]]

我搞不懂。

Answer 1

如果分组是由每个列表中的前三个元素决定的，下面的代码将满足您的要求：

from collections import defaultdict

example_list=[["a","b","c", 2,4,5,7],["e","f","g",0,0,1,5],["e","f","g", 1,4,5,7],["a","b","c", 3,2,5,7]]
d = defaultdict(list)
for l in example_list:
    d[tuple(l[:3])].append(l)

print d.values() # [[['a', 'b', 'c', 2, 4, 5, 7], ['a', 'b', 'c', 3, 2, 5, 7]], ...]

这将使用defaultdict生成一个字典，其中键是前三个元素，值是从这些元素开始的列表列表。

Answer 2

首先，简单地使用sorted()对列表进行排序，提供一个lambda函数作为键。

>>> a = sorted(example_list, key=lambda x:x[:3])
[['a', 'b', 'c', 2, 4, 5, 7], ['a', 'b', 'c', 3, 2, 5, 7], ['e', 'f', 'g', 0, 0, 1, 5], ['e', 'f', 'g', 1, 4, 5, 7]]

然后在排序列表上使用itertools.groupby()：

>>> [list(v) for k, v in groupby(a, lambda x:x[:3])]
[
    [['a', 'b', 'c', 2, 4, 5, 7], ['a', 'b', 'c', 3, 2, 5, 7]], 
    [['e', 'f', 'g', 0, 0, 1, 5], ['e', 'f', 'g', 1, 4, 5, 7]]
]