Python 之 列表推导式

简单的例子

list1 = [1,2,3,4,5,6]
list2 = []
for num in list1:
    list2.append(num+1)
print(list2)

运行结果：

使用列表推导式，直接将for循环语句写入列表当中

list1 = [1,2,3,4,5,6]
list2 = [num+1 for num in list1]
print(list2)

运行结果不变：

高级用法的例子

使用列表推导式，可以在for语句后面跟上一个if判断语句，用于过滤掉那些不满足条件的结果项。

line1 = "薛长江    4560 25"
parts = line1.split(' ')
print(parts)
line2 = [i for i in parts if i]
print(line2)

运行结果：

for 循环后可以嵌套for循环。

lis = [(i, j) for i in range(2) for j in range(1, 3)]
print(lis)

运行结果：