您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

Python根据模式将列表拆分为子集

Python根据模式将列表拆分为子集

您可以使用itertools.groupby

>>> import itertools
>>> mylist = ['sub_0_a', 'sub_0_b', 'sub_1_a', 'sub_1_b']
>>> for k,v in itertools.groupby(mylist,key=lambda x:x[:5]):
...     print k, list(v)
... 
sub_0 ['sub_0_a', 'sub_0_b']
sub_1 ['sub_1_a', 'sub_1_b']

或完全按照您的指定:

>>> [list(v) for k,v in itertools.groupby(mylist,key=lambda x:x[:5])]
[['sub_0_a', 'sub_0_b'], ['sub_1_a', 'sub_1_b']]

当然,需要注意一些常见的警告(请确保您的列表使用与分组时使用的键相同的键排序),并且您可能需要稍微复杂一些的键功能才能获取真实的数据…

python 2022/1/1 18:50:38 有357人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶