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

Question

输入

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

期望的输出:

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

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

Answer 1

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]