问题描述
对于以下列表:
races=['R1','R2','R3','R4','R1','R5']
我想将其分为:
list_1=['R1','R4']
list_2=['R1','R2']
list_3=['R1','R5']
所以-在使用列表理解时,想到“ R”数字开始递减后,就会开始一个新列表。 我对list_1的尝试是:
list_1=[i for i in races if int(i.split('R')[1])>int((i-1).split('R')[1])]
这不太有效-以及如何处理list_2和list_3?
解决方法
如果您尝试使用列表推导,您将使其变得更加复杂。显式循环提供了一种简单的解决方案:
races = ['R1','R2','R3','R4','R1','R5']
out = []
for r in races:
if not out or r < out[-1][-1]:
out.append([])
out[-1].append(r)
print(out)
礼物:
[['R1','R4'],['R1','R2'],'R5']]
请注意,实际拆分为单个变量将需要执行一些“ hacky”操作(例如使用globals
或exec
)。如示例所示,将输出保存在一个集合中,并根据需要建立索引。
但是,在已知输出变量数量的特殊情况下,您可以将其解压缩:
list_1,list_2,list_3 = out