2

我有一份表格清单：

>>> my_list = ['BLA1', 'BLA2', 'BLA3', 'ELE1', 'ELE2', 'ELE3', 'PRI1', 'PRI2', 'NEA1', 'NEA2', 'MAU1', 'MAU2', 'MAU3']

我想创建一个新列表，将重复的元素分组到我的新列表中的列表中，所以最后我将拥有：

>>> new_list = [['BLA1', 'BLA2', 'BLA3'], ['ELE1', 'ELE2', 'ELE3'], ['PRI1', 'PRI2'], ['NEA1', 'NEA2'], ['MAU1', 'MAU2', 'MAU3']]
4

3 回答 3

6

使用itertools.groupby：

import itertools

[list(group) for key, group in itertools.groupby(my_list, key=lambda v: v[:3])]

这里需要该key参数来提取您想要分组的值的一部分；前 3 个字符。

结果：

>>> my_list = ['BLA1', 'BLA2', 'BLA3', 'ELE1', 'ELE2', 'ELE3', 'PRI1', 'PRI2', 'NEA1', 'NEA2', 'MAU1', 'MAU2', 'MAU3']
>>> [list(group) for key, group in itertools.groupby(my_list, key=lambda v: v[:3])]
[['BLA1', 'BLA2', 'BLA3'], ['ELE1', 'ELE2', 'ELE3'], ['PRI1', 'PRI2'], ['NEA1', 'NEA2'], ['MAU1', 'MAU2', 'MAU3']]

groupby会将相等的连续键组合成 1 组。如果您有不相交的组（相同的值，但中间有其他值），它将为这些组创建单独的组：

>>> my_list = ['a1', 'a2', 'b1', 'b2', 'a3', 'a4']
>>> [list(group) for key, group in itertools.groupby(my_list)]
[['a1', 'a2'], ['b1', 'b2'], ['a3', 'a4']]

如果这不是你想要的，你必须先排序my_list。

于 2013-03-26T13:06:56.473 回答
1

确保已分类并使用

itertools.groupy
于 2013-03-26T13:06:05.543 回答
1

作为 groupby 的替代方案，您可以使用collections.Counter：

In [40]: from collections import Counter

In [41]: [ [k]*v for (k,v) in Counter(my_list).iteritems() ]
Out[41]: 
[['PRI', 'PRI'],
 ['NEA', 'NEA'],
 ['BLA', 'BLA', 'BLA'],
 ['MAU', 'MAU', 'MAU'],
 ['ELE', 'ELE', 'ELE']]

与 groupby 不同，如果元素全部混乱，这将无需对列表进行排序。

于 2013-03-26T13:23:00.160 回答