将两个嵌套for循环写成一个列表生成式
如,有一个嵌套列表,a=[[1,2],[3,4],[5,6]],要提取列表里的每一个元素
用for循环处理:
for i in a:
for j in i:
print(j)
用列表生成式:
b=[j for i in a for j in i]#注意两个for的顺序
print(b)
>>> b
[1, 2, 3, 4, 5, 6]
将两个嵌套for循环写成一个列表生成式
如,有一个嵌套列表,a=[[1,2],[3,4],[5,6]],要提取列表里的每一个元素
用for循环处理:
for i in a:
for j in i:
print(j)
用列表生成式:
b=[j for i in a for j in i]#注意两个for的顺序
print(b)
>>> b
[1, 2, 3, 4, 5, 6]