如何解决如何在条件下拆分元组列表?
我是Python的新手,所以这里的任务很简单。有谁知道解决问题的最优雅方式?
例如,我有一个元组列表:
my_list = [(1,'a','b','c'),(2,'d','e','f'),(3,'g','h','i'),(1,'j','k','l'),'m','n','o'),'p','q','r'),'s','t','u')]
我需要将其拆分为元组的子列表,以第一项为“ 1”的元组开始。
[(1,'i')]
[(1,'o')]
[(1,'u')]
解决方法
您正在有效地计算某种“ groupwhile”函数-您想对从1
开始的每个元组进行拆分。这看起来很像itertools.groupby
,如果我们保留一小部分的全局状态(在我们的示例中为one_count
变量),我们可以重新使用已经内置于其中的分组/聚合逻辑语言来获得您想要的结果。
import itertools
# The inner function is just so that one_count will be initialized only
# as many times as we want to call this rather than exactly once via
# some kind of global variable.
def gen_count():
def _cnt(t,one_count=[0]):
if t[0] == 1:
one_count[0] += 1
return one_count[0]
return _cnt
result = [list(g[1]) for g in itertools.groupby(my_list,key=gen_count())]
更传统的解决方案是遍历您的示例并将中间输出附加到结果集。
result = []
for i,*x in my_list:
if i==1:
result.append([(i,*x)])
else:
result[-1].append((i,*x))
,
尝试此代码。我认为中断是在再次找到第一个字符(1)时。我还假设输出是一个列表。
my_list = [(1,'a','b','c'),(2,'d','e','f'),(3,'g','h','i'),(1,'j','k','l'),'m','n','o'),'p','q','r'),'s','t','u')]
ch = my_list[0][0]
all = []
st = 0
for i,t in enumerate(my_list):
if t[0] == ch:
if i != 0:
all.append(my_list[st:i])
st = i
else:
all.append(my_list[st:i])
print(all)
输出
[
[(1,'i')],[(1,'o')],'r')]
]
版权声明:本文内容由互联网用户自发贡献,该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 dio@foxmail.com 举报,一经查实,本站将立刻删除。