python学习之路(13)
列表生成式
列表生成式即List Comprehensions,是Python内置的非常简单却强大的可以用来创建list的生成式。
举个例子,要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
可以用range(1, 11)
:
>>> range(1,11)
range(1, 11
但如果要生成[1x1, 2x2, 3x3, ..., 10x10]
怎么做?方法一是循环:
>>> range(1,11) range(1, 11) >>> list=[] >>> for x in range(1,11): list.append(x*x) >>> list [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
但是循环太繁琐,而列表生成式则可以用一行语句代替循环生成上面的list:
>>> list=[x*x for x in range(1,11) if x%2==0] >>> list [4, 16, 36, 64, 100]
还可以使用两层循环,可以生成全排列:
>>> list=[m+n for m in'dasd' for n in'sad'] >>> list ['ds', 'da', 'dd', 'as', 'aa', 'ad', 'ss', 'sa', 'sd', 'ds', 'da', 'dd']
三层和三层以上的循环就很少用到了。
运用列表生成式,可以写出非常简洁的代码。例如,列出当前目录下的所有文件和目录名,可以通过一行代码实现:
>>> import os >>> list=[d for d in os.listdir('.')] >>> list ['10.py', '360安全卫士.lnk', '7dLauncher.exe', '8月新答题', 'chenzaifen', 'cmd.exe.lnk', 'desktop.ini', 'Dota 2.url', 'Google Chrome.lnk', 'mimikatz_trunk.rar', 'phpStudy.lnk', 'pop3', 'Python电子邮件', 'Python电子邮件.zip', 'WeGame.lnk', 'wire', '~$PowerShell.exe 执行 Empire 的几种姿势.doc', '助教工作', '协议分析', '学习之路', '密码学重修', '小工具', '新建文件夹', '新建文件夹 (2)', '新建文件夹 (3)', '极速PDF编辑器.lnk', '爱奇艺.lnk', '百度网盘.lnk', '网络协议', '网络渗透', '逆向', '酷狗音乐.lnk']
for
循环其实可以同时使用两个甚至多个变量,比如dict
的iteritems()
可以同时迭代key和value:
>>> for k,v in d.items(): print(k,'=',v) sdad = dsad adsa = sdaasda asdassad = asdas
因此,列表生成式也可以使用两个变量来生成list:
>>> d={'x':'1','y':'2','z':'3'} >>> list=[k+'='+v for k,v in d.items()] >>> list ['x=1', 'y=2', 'z=3']
最后把一个list中所有的字符串变成小写:
>>> list=['DSADA','DSADSA','SA'] >>> list=[s.lower()for s in list] >>> list ['dsada', 'dsadsa', 'sa']
运用列表生成式,可以快速生成list,可以通过一个list推导出另一个list,而代码却十分简洁。
思考:如果list中既包含字符串,又包含整数,由于非字符串类型没有lower()
方法,所以列表生成式会报错:
>>> L= ['Hello', 'World', 18, 'Apple', None] >>> [s.lower() for s in L] Traceback (most recent call last): File "<pyshell#106>", line 1, in <module> [s.lower() for s in L] File "<pyshell#106>", line 1, in <listcomp> [s.lower() for s in L] AttributeError: 'int' object has no attribute 'lower'
使用内建的isinstance
函数可以判断一个变量是不是字符串:
>>> x='dasd' >>> y=123151 >>> isinstance(x,str) True >>> isinstance(y,str) False