首页 > 代码库 > 迭代与列表生成式、生成器
迭代与列表生成式、生成器
一、迭代
python中如何判断一个对象是可迭代对象呢?方法是通过collections模块的Iterable类型判断:
>>> from collections import Iterable>>> isinstance(‘abc‘, Iterable) # str是否可迭代True>>> isinstance([1,2,3], Iterable) # list是否可迭代True>>> isinstance(123, Iterable) # 整数是否可迭代False
Python内置的enumerate函数可以把一个list变成索引-元素对,这样就可以在for循环中同时迭代索引和元素本身:
>>> for i, value in enumerate([‘A‘, ‘B‘, ‘C‘]):... print i, value...0 A1 B2 C
>>> for x, y in [(1, 1), (2, 4), (3, 9)]:... print x, y...1 12 43 9
>>> d = {‘x‘: ‘A‘, ‘y‘: ‘B‘, ‘z‘: ‘C‘ }>>> for k, v in d.iteritems():... print k, ‘=‘, v... y = Bx = Az = C
二、列表生成式
要生成list [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]可以用range(1, 11):
>>> range(1, 11)[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
如果要生成[1x1, 2x2, 3x3, ..., 10x10]怎么做?方法一是循环:
如果要生成[1x1, 2x2, 3x3, ..., 10x10]怎么做?方法一是循环:
>>> L = []>>> for x in range(1, 11):... L.append(x * x)...>>> L[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
如果要生成[1x1, 2x2, 3x3, ..., 10x10]怎么做?方法二是列表生成式:
>>> [x * x for x in range(1, 11)][1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
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
>>> [d for d in os.listdir(‘.‘)]# os.listdir可以列出文件和目录
[‘DLLs‘, ‘Doc‘, ‘geckodriver-v0.11.1-win64.zip‘, ‘geckodriver.exe‘, ‘include‘, ‘Lib‘, ‘libs‘, ‘pip-8.1.2‘, ‘pip-8.1.2.tar.gz‘, ‘python.exe‘, ‘python2.7.exe‘, ‘python2.exe‘, ‘pythonw.exe‘, ‘pythonw2.7.exe‘, ‘pythonw2.exe‘, ‘Scripts‘, ‘selenium‘, ‘setuptools-28.7.1‘, ‘setuptools-28.7.1.zip‘, ‘tcl‘, ‘Tools‘, ‘w9xpopen.exe‘]
列表生成式也可以使用两个变量来生成list:
>>> d = {‘x‘: ‘A‘, ‘y‘: ‘B‘, ‘z‘: ‘C‘ }>>> [k + ‘=‘ + v for k, v in d.iteritems()][‘y=B‘, ‘x=A‘, ‘z=C‘]
三、生成器
在Python中,一边循环一边计算的的机制称为生成器(Generator)
1.创建Generator,只要把一个列表生成式的[]改成(),如下:
>>> L = [x * x for x in range(10)]>>> L[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]>>> g = (x * x for x in range(10))>>> g<generator object <genexpr> at 0x104feab40>
2.打印Generator的元素,可以通过generator的next()方法
>>> g.next()0>>> g.next()1>>> g.next()4>>> g.next()9>>> g.next()16>>> g.next()25>>> g.next()36>>> g.next()49>>> g.next()64>>> g.next()81>>> g.next()Traceback (most recent call last): File "<stdin>", line 1, in <module>StopIteration
generator保存的是算法,每次调用next(),就计算出下一个元素的值,直到计算到最后一个元素,没有更多的元素时,抛出StopIteration的错误,
还可使用for循环,因为generator也是可迭代对象,如下:
>>> g = (x * x for x in range(10))>>> for n in g:... print n...0149162536496481
迭代与列表生成式、生成器
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。