python之按特定元素拆分列表以获取列表列表 Python

输入

['~', 'n1', 'n2', ..., 'nn', '~', 'k1', 'k2', ..., 'kn', '~'] 

期望的输出:

[['n1', 'n2', ..., 'nn'],['k1', 'k2', ..., 'kn']] 

我见过 itertools groupby 但无法让它工作。任何帮助，将不胜感激。而且 ... 实际上不在列表中，只是说中间还有更多元素

请您参考如下方法:

groupby 方法是最好的方法。

使用键函数 item != '~' 进行分组，并过滤​​为 True 的键(当 x=='~'，关键函数返回 False，if k 条件将其过滤掉)

import itertools 
 
lst = ['~', 'n1', 'n2', 'nn', '~', 'k1', 'k2', 'kn', '~'] 
 
result = [list(v) for k,v in itertools.groupby(lst,lambda x : x!='~') if k] 

结果:

[['n1', 'n2', 'nn'], ['k1', 'k2', 'kn']] 

请注意，您必须对发布的组强制迭代，因为 groupby 返回可迭代对象(以防万一您只需要再次迭代它们)

如果你有空字符串，那就更简单了:不需要 lambda，依赖值的真实性并使用 bool 运算符:

lst = ['', 'n1', 'n2', 'nn', '', 'k1', 'k2', 'kn', ''] 
 
result = [list(v) for k,v in itertools.groupby(lst,bool) if k]