小志的Python 学习笔记（二）

python列表推导式

示例：

>>>alist=[x*x for x in range(10)]

>>>alist

[0,1,4,9,16,25,36,49,64,81]

 

相当于

>>>alist=[]

>>> for x in range (10):

alist.append(x * x)

2. 使用列表推导式实现嵌套列表的平铺

>>>vec=[[1,2,3],[4,5,6],[7,8,9]]

>>>[num for elem in evc for num in elem]

[1,2,3,4,5,6,7,8,9]