【Python】学习笔记之列表生成式

列表生成式

主要用于生成较为复杂的列表

常用用法

>>> [x * x for x in range(5) if x % 3 !=1 ]
[0, 4, 9]   #返回除以3余数不为1的数字的平方

两层循环

可以生成全排列:

>>> [m + n for m in '123' for n in 'abc']
['1a', '1b', '1c', '2a', '2b', '2c', '3a', '3b', '3c']

实例

运用列表生成式,可以写出非常简洁的代码。例如,列出当前目录下的所有文件和目录名,可以通过一行代码实现:

>>> os.getcwd()
'D:\\python'
>>> [d for d in os.listdir(os.getcwd())]  #列出当前目录下的文件和目录
['DLLs', 'Doc', 'geckodriver.exe', 'include', 'Lib', 'libs', 'LICENSE.txt', 'NEWS.txt', 'python.exe', 'pythonw.exe', 'README.txt', 'Scripts', 'selenium', 'tcl', 'Tools', 'w9xpopen.exe']

for循环其实可以同时使用两个甚至多个变量,比如dictitems()可以同时迭代key和value:

>>> for k ,v in a.items():
    print k ,'=' ,v
a = 1
b = 2

因此,列表生成式也可以使用两个变量来生成list:

>>> [k+'='+str(v) for k,v in a.items()]
['a=1', 'b=2']
>>> [k.upper()+'='+str(v) for k,v in a.items() if v != 1]  #加入判断和其他表达式
['B=2']

小结

运用列表生成式,可以快速生成list

上一篇:java实现Kafka生产者示例


下一篇:Oracle查询优化-插入、更新与删除