Python: 列表推导式--轻量级循环

时间:2023-01-18 15:40:12

定义: 列表推导式(list comprehension)是利用其他列表创建新列表的一种方法,其工作方式类似于for循环,对列表进行过滤变种操作

eg1:

>>> [x*x for x in range(10)]
 [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]

eg2:

>>> mylist = [1, 4, -5, 10, -7, 2, 3, -1]
>>> import math
>>> [math.sqrt(n) for n in mylist if n > 0]
[1.0, 2.0, 3.1622776601683795, 1.4142135623730951, 1.7320508075688772]

过滤操作的一个变种就是将不符合条件的值用新的值代替。在一列数据中你可能不仅想找到正数,还想将不是正数的数替换成指定的数。通过将过滤条件放到条件表达式中去,就像这样:
>>> clip_neg = [n if n > 0 else 0 for n in mylist]
>>> clip_neg
[1, 4, 0, 10, 0, 2, 3, 0]