熟悉python的朋友都知道,列表生成器很好用,但有时搞不清楚怎么用. 列表生成器和生成器generator的关系就不细说了.此处主要帮助大家理解.
如下面的列表生成器
G = [xx for x in range(5)]
xx是表达式,是我们想如何操作, 可以理解成:
for x in range(5):
x*x
将这个for循环的结果放到列表中. [0, 1, 4, 9, 16]
也可以有两个变量:
G = [x*y for x in range(1,5) for y in range(2,4) if x%2==0] 类似于以下语句:
for x in range(1,5):
for y in range(2,4):
if x%2==0:
x*y
将这个for循环的结果放到列表中: [4, 6, 8, 12]
也可以用列表生成器产生多维数组:
G = [[x,y,z, x+y+z] for x in range(1,5) for y in range(2,4) for z in range(7,9)]
产生的列表为:
[[1, 2, 7, 10], [1, 2, 8, 11], [1, 3, 7, 11], [1, 3, 8, 12], [2, 2, 7, 11], [2, 2, 8, 12], [2, 3, 7, 12], [2, 3, 8, 13], [3, 2, 7, 12], [3, 2, 8, 13], [3, 3, 7, 13], [3, 3, 8, 14], [4, 2, 7, 13], [4, 2, 8, 14], [4, 3, 7, 14], [4, 3, 8, 15]]