Python：减少列表但保留细节

Question

假设我有一个项目列表，其中一些项目在某种程度上相似 但在一个点后相差一个数字

['abc.1',
 'abc.2',
 'abc.3',
 'abc.7',
 'xyz.1',
 'xyz.3',
 'xyz.11',
 'ghj.1',
 'thj.1']

我想从这个列表中生成一个新列表，它可以折叠多个但保留它们的一些数据，即数字后缀 所以上面的列表应该产生一个新的列表

[('abc',('1','2','3','7'))
 ('xyz',('1','3','11'))
 ('ghj',('1'))
 ('thj',('1'))]

我的想法是，第一个列表可以通过点分成对 但是然后我如何按第一部分对对进行分组而不丢失第二部分

对不起，如果这个问题很无聊，提前谢谢

... 哇，没想到这么快就有这么多好答案，谢谢

Answer 1

from collections import defaultdict

d = defaultdict(list)

for el in elements:
    key, nr = el.split(".")
    d[key].append(nr)

#revert dict to list
newlist = d.items()

Answer 2

使用分隔函数映射列表，使用带有第一个元素的键的itertools.groupby，并将第二个元素收集到结果中。

from itertools import groupby, imap

list1 = ["abc.1", "abc.2", "abc.3", "abc.7", "xyz.1", "xyz.3", "xyz.11", "ghj.1", "thj.1"]

def break_up(s):
    a, b = s.split(".")
    return a, int(b)

def prefix(broken_up): return broken_up[0]
def suffix(broken_up): return broken_up[1]

result = []
for key, sub in groupby(imap(break_up, list1), prefix):
    result.append((key, tuple(imap(suffix, sub))))

print result

输出：

[('abc', (1, 2, 3, 7)), ('xyz', (1, 3, 11)), ('ghj', (1,)), ('thj', (1,))]