迭代器和可迭代協(xié)議

可迭代對(duì)象包括實(shí)際序列和按照需求計(jì)算的虛擬序列

#手動(dòng)迭代：iter和next

#python3.0提供了一個(gè)內(nèi)置函數(shù)next，它會(huì)自動(dòng)調(diào)用一個(gè)對(duì)象的next方法，給定一個(gè)
#可迭代對(duì)象x。調(diào)用next(x)等同于x,next()

列表以及其他對(duì)象不是自身的迭代器，他們可以調(diào)用iter來(lái)啟動(dòng)迭代

l = [1, 2, 3]
print(iter(l) is l)

False

#print(l.next())

AttributeError: 'list' object has no attribute 'next'

l = [1, 2, 3]
i = iter(l)
print(next(i))

1

print(i.next())

2

l = [1, 2, 3]
for x in l:
print(x**2, end=' ')

1 4 9

l = [1, 2, 3]
i = iter(l)
while True:
try:
j = next(i)
except StopIteration:
break
else:
print(j ** 2, end=' ')

1 4 9

python2 中迭代方法是x.next, python3是x.next,但next函數(shù)確是都可用的。

其他迭代器

d = {'a':1, 'b':2, 'c':3}
for k in d.keys():
print(k, d[k])

a 1

b 2

c 3

#python3中，字典自己有個(gè)迭代器
i = iter(d)
print(next(i))
print(next(i))
print(next(i))

a

b

c

因此我們不需要調(diào)用keys方法來(lái)遍歷字典，for將使用迭代協(xié)議獲取鍵。

for k in d:
print(k, d[k])

a 1

b 2

c 3

shelves和os.popen的結(jié)果也是可迭代的

import os
p = os.popen('dir')
print(p.next())

驅(qū)動(dòng)器 F 中的卷沒(méi)有標(biāo)簽。

注意py2中popen對(duì)象支持p.next方法，py3中支持p.next()方法。

#迭代協(xié)議也是需要將結(jié)果包裝到一個(gè)list中才能看到值?？傻鷮?duì)象一次返回一

個(gè)結(jié)果，而不是一個(gè)列表

a = range(5)
print(a)

range(0, 5)

i = iter(a)
print(next(i))
print(next(i))

0

1

print(list(range(5)))

[0, 1, 2, 3, 4]

列表解析

l = []
for i in [1, 2, 3]:
l.append(i+1)
print(l)

[2, 3, 4]

相比普通的循環(huán)，列表解析也可以達(dá)到相同的目的，并且更加簡(jiǎn)潔，運(yùn)行速度也比

普通的循環(huán)快

print([i+1 for i in [1, 2, 3]])

[2, 3, 4]

文件中的列表解析

with open('some.py', 'w') as f:
f.write('import os\n')
f.write('a = 1\n')
f.write('b = 2\n')
with open('some.py') as f:
print(f.read())

import os

a = 1

b = 2

f = open('some.py')
line = f.readlines()
print(line)

['import os\n', 'a = 1\n', 'b = 2\n']

如果要去掉行末的換行符的話。

print([l.rstrip() for l in line])

['import os', 'a = 1', 'b = 2']

因?yàn)榱斜斫馕鱿駀or循環(huán)語(yǔ)句是一個(gè)迭代環(huán)境， 我們甚至不用提前打開(kāi)文件。

l = [line.rstrip() for line in open('some.py')]
print(l)

['import os', 'a = 1', 'b = 2']

除此之外。列表解析的表現(xiàn)力也很強(qiáng)，我們可以在迭代時(shí)在一個(gè)文件上運(yùn)行任何的

#字符串操作
print([line.upper() for line in open('some.py')])

['IMPORT OS\n', 'A = 1\n', 'B = 2\n']

列表解析也可以嵌套for循環(huán)和if

lines = [line.rstrip() for line in open('some.py') if line[0] == 'a']
print(lines)

['a = 1']

print([x+y for x in 'abc' for y in 'lmn'])

['al', 'am', 'an', 'bl', 'bm', 'bn', 'cl', 'cm', 'cn']

#列表解析，in成員測(cè)試，map內(nèi)置函數(shù)，sorted和zip都使用了迭代協(xié)議
#當(dāng)應(yīng)用文件時(shí)，會(huì)自動(dòng)掃描

upper = [line.upper() for line in open('some.py')]
print(upper)

['IMPORT OS\n', 'A = 1\n', 'B = 2\n']

print(list(map(str.upper, open('some.py'))))

['IMPORT OS\n', 'A = 1\n', 'B = 2\n']

print('a = 1\n' in open('some.py'))

True

print(sorted(open('some.py')))

['a = 1\n', 'b = 2\n', 'import os\n']

print(list(zip(open('some.py'), open('some.py'))))

[('import os\n', 'import os\n'), ('a = 1\n', 'a = 1\n'), ('b = 2\n', 'b = 2\n')]

print(list(enumerate(open('some.py'))))

[(0, 'import os\n'), (1, 'a = 1\n'), (2, 'b = 2\n')]

print(list(filter(bool, open('some.py'))))

['import os\n', 'a = 1\n', 'b = 2\n']

import functools, operator
print(functools.reduce(operator.add, open('some.py')))

import os

a = 1

b = 2

print(sum([1, 2, 3, 4, 5]))

15

print(any(['spam', '', 'mi']))

True

print(all(['spam', '', 'mi']))

False

print(max([1, 2, 3, 4, 5]))

5

print(min([1, 2, 3, 4, 5]))

1

max 和min 也可以應(yīng)用于文件

print(max(open('some.py')))

import os

print(min(open('some.py')))

a = 1

迭代協(xié)議也可直接作用于文件

print(list(open('some.py')))

['import os\n', 'a = 1\n', 'b = 2\n']

print(tuple(open('some.py')))

('import os\n', 'a = 1\n', 'b = 2\n')

print('&&'.join(open('some.py')))# (牛逼)
#import os
#&&a = 1
#&&b = 2

a, *b = open('some.py')
print(a)
print(b)
#import os
#
#['a = 1\n', 'b = 2\n']

解析和集合

print(set(open('some.py')))

{'b = 2\n', 'a = 1\n', 'import os\n'}

print({line for line in open('some.py')})

{'a = 1\n', 'b = 2\n', 'import os\n'}

print({ix: line for ix, line in enumerate(open('some.py'))})

{0: 'import os\n', 1: 'a = 1\n', 2: 'b = 2\n'}

print({line for line in open('some.py') if line[0] == 'a'})

{'a = 1\n'}

print({ix: line for ix, line in enumerate(open('some.py')) if line[0] == 'a'})

{1: 'a = 1\n'}

def f(a, b, c):
print(a, b, c, sep = '&')

f(1, 2, 3)

1-2-3

f(*open('some.py'))

import os

&a = 1

&b = 2

x , y = (1, 2), (3, 4)
print(list(zip(x, y)))

[(1, 3), (2, 4)]

a, b = zip(*zip(x, y))
print(a)
print(b)

(1, 2)

(3, 4)

range和其他迭代器不同，rang它不是自己的迭代器，并且他支持在結(jié)果上的多個(gè)迭代器

他們會(huì)記住自己的位置

r = range(3)
i1 = iter(r)
print(next(i1))

0

print(next(i1))

1

i2 = iter(r)
print(next(i2))

0

print(next(i1))

2

而zip，map,filter不支持相同結(jié)果上的活躍迭代器

z = zip((1, 2, 3), (10, 11, 12))
i1 = iter(z)
i2 = iter(z)
print(next(i1))

(1, 10)

print(next(i1))

(2, 11)

print(next(i2))

(3, 12)

m = map(abs,(-1, 0, 1))
i1 = iter(m)
i2 = iter(m)
print(next(i1)) # 1
print(next(i1)) # 0
print(next(i1)) # 1
#print(next(i2)) # StopIteration

r = range(3)
r1, r2 = iter(r), iter(r)
print(next(r1)) # 0
print(next(r1)) # 1
print(next(r1)) # 2
print(next(r2)) # 0

通過(guò)iter返回一個(gè)新的對(duì)象來(lái)支持多個(gè)迭代器，單個(gè)迭代器返回自身。

py3中字典中由于keys不在返回一個(gè)列表，所以應(yīng)該首先用list來(lái)展示他

或者使用sorted

d = {'a':1, 'b':2, 'c':3}
for k in sorted(d.keys()):
print(k, d[k], end=' ')

a 1 b 2 c 3

for k in sorted(d):
print(k, d[k], end = ' ')

a 1 b 2 c 3