列表生成式即List Comprehensions,是Python内置的非常简单却强大的可以用来创建list的生成式
- 例如生成[1,2,3,4,5,6,7,8,9,10]:
>>> list(range(1,11))
[1,2,3,4,5,6,7,8,9,10]
- 生成[11, 22, 33, ..., 1010]:
>>> [x*x for x in range(1,11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
写列表生成式时,把要生成的元素x * x放到前面,后面跟for循环,就可以把list创建出来
- for循环后面还可以加上if判断,这样我们就可以筛选出仅偶数的平方:
>>> [x*x for x in range(1, 11) if x%2 ==0 ]
[4,16,36, 64, 100]
- 还可以使用两层循环,可以生成全排列
>>> [m+n for m in 'abc' for n in 'xyz']
['ax', 'ay', 'az', 'bx', 'by', 'bz', 'cx', 'cy', 'cz']
- 列出当前目录下的所有文件和目录名,可以通过一行代码实现
>>> import os # 导入os模块
>>> [d for d in os.listdir('.')]
['.emacs.d', '.ssh', '.Trash', 'Adlm', 'Applications', 'Desktop', 'Documents', 'Downloads', 'Library', 'Movies', 'Music', 'Pictures', 'Public', 'VirtualBox VMs', 'Workspace', 'XCode']
- 列表生成式也可以使用2个变量来生成list:
>>> d={'x':'a', 'y':'b', 'z':'c'}
>>> [ k+ '=' + v for k,v in d.items()]
['y=b', 'x=a', 'z=c']
- 把list中的所有字符串变成小写
>>> L=['Hello', 'World', 'IBM', 'Apple']
>>> [s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']
- 若list中含有字符串,数字,则可以:
>>> L = ['Hello', 'World', 18, 'Apple']
>>> [d.lower() for d in L if isinstance(d, str)]
['hello', 'world', 'apple']