将两个嵌套for循环写成一个列表生成式

来源:互联网 发布:python 加减乘除 编辑:程序博客网 时间:2024/05/17 02:55

将两个嵌套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]