如何根据某些条件迭代1d列表时创建多维列表。
我正在迭代一个列表,每当我找到一个' \ n'我应该使用新列表附加如此创建的列表,例如
set term
所以我希望它成为,
a = [1,2,3,4,5,'\n',6,7,8,9,0,'\n',3,45,6,7,2]
我应该怎么做?请帮忙
new_list = [[1,2,3,4],[6,7,8,9,0],[3,45,6,7,2]]
这就是我的尝试。
答案 0 :(得分:3)
示例代码:
lst = [[]]
for x in a:
if x != '\n':
lst[-1].append(x)
else:
lst.append([])
print(lst)
输出:
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 0], [3, 45, 6, 7, 2]]
答案 1 :(得分:1)
使用itertools.groupby
可以完成这项工作(按分组作为换行符):
import itertools
a = [1,2,3,4,5,'\n',6,7,8,9,0,'\n',3,45,6,7,2]
new_list = [list(x) for k,x in itertools.groupby(a,key=lambda x : x!='\n') if k]
print(new_list)
我们比较关键真值以过滤掉\n
结果:
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 0], [3, 45, 6, 7, 2]]
答案 2 :(得分:0)
以下是基本方法:
编辑:天哪!傻傻的虫......这是一个修复:
>>> sub = []
>>> final = []
>>> for e in a:
... if e == '\n':
... final.append(sub)
... sub = []
... else:
... sub.append(e)
... else:
... final.append(sub)
...
>>> final
[[1, 2, 3, 4, 5], [6, 7, 8, 9, 0], [3, 45, 6, 7, 2]]
>>>
还有其他方法,但这是天真的必要方式。
答案 3 :(得分:0)
我就是这样做的,但必须有更好的解决方案。
x = 0
output = [[]]
for item in a:
if item != "\n":
output[x].append(item)
else:
x += 1
output.append([])
print(output)