按每个元素中的字符拆分python列表

Question

我有一个python列表,其中的元素是一个包含字母和数字字符串的列表,我想知道如何将字符串拆分为字符串开头的字符,而不为每个字符创建一个语句.所以我想将mylist拆分为列表a,b和c.

mylist = [['a1'],['a2'],['c1'],['b1']]
a = [['a1'],['a2']]
b = [['b1']]
c = [['c1']]

重要的是我将它们保存为列表列表(即使它只是每个小列表中的一个元素).

谢谢

Answer 1

这将起作用：

import itertools as it

mylist  = [['a1'],['a2'],['c1'],['b1']]
keyfunc = lambda x: x[0][0]

mylist = sorted(mylist, key=keyfunc)
a, b, c = [list(g) for k, g in it.groupby(mylist, keyfunc)]

sorted()仅当 inmylist中的元素尚未按字符串开头的字符排序时，才需要使用where 行。

编辑 ：

正如评论中所指出的，更通用的解决方案（不将变量数量限制为三个）将使用字典推导（在 Python 2.7+ 中可用），如下所示：

result_dict = {k: list(g) for k, g in it.groupby(mylist, keyfunc)}

现在答案在字典中按第一个字符键入：

result_dict['a']
> [['a1'],['a2']]

result_dict['b']
> [['b1']]

result_dict['c']
> [['c1']]