列表推导是一种从其他列表创建列表的方式,类似于数学中的集合推导。
列表推导的使用方式非常简单,有点类似于for循环。
>>> [x * x for x in range(10)] [0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
如果想打印那些能被3整除的平方值呢?这时候可以在列表推导中添加一条if语句:
>>> [x * x for x in range(10) if x % 3 == 0] [0, 9, 36, 81]
还可添加多层for循环(三层和三层以上的循环很少用到):
>>> [(x, y) for x in range(3) for y in range(3)] [(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]
运用列表生成式,可以写出非常简洁的代码。
>>> d = {'x': 'A', 'y': 'B', 'z': 'C' } >>> [k + '=' + v for k, v in d.items()] ['x=A', 'y=B', 'z=C']
>>> L = ['Hello', 'World', 'IBM', 'Apple'] >>> [s.lower() for s in L] ['hello', 'world', 'ibm', 'apple']