python - 如何根据项目的第一个字符对可迭代对象中的项目进行分组？

Question

从一个已排序的可迭代对象开始，我需要按项目的第一个字符对项目进行分组(比如从 a 到 z 的每个字母一组，数字和符号一组)。

举个更具体的例子，假设我有这个列表:

L = ['aa', 'ab', 'ac', 'ba', 'bb', 'bc', 'ca', 'cb', 'cc', '10', '%a' , 'b', ...]

我需要这样的东西:

GL = [['aa', 'ab', 'ac'], ['ba', 'bb', 'bc'], ['ca', 'cb', 'cc'], [' 10', '%a', 'b']]

这样做有哪些选择，哪个最有效？

Answer 1

import itertools as it

L = ['aa', 'ab', 'ac', 'ba', 'bb', 'bc', 'ca', 'cb', 'cc', '10', '%a', ' b']

sorter = lambda x: x[0].lower() if x and x[0].isalpha() else '}'
GL = [list(v) for k, v in it.groupby(sorted(L, key=sorter), key=sorter)]

返回:

[['aa', 'ab', 'ac'],
 ['ba', 'bb', 'bc'],
 ['ca', 'cb', 'cc'],
 ['10', '%a', ' b']]

您可以使用其他东西代替 '}' 将所有非字母字符放在排序中的正确位置。