切片
# L[0:3]表示,从索引0开始取,直到索引3为止,但不包括索引3
>>> L = ['Michael', 'Sarah', 'Tracy', 'Bob', 'Jack']
>>> L[0:3]
['Michael', 'Sarah', 'Tracy']
# 如果第一个索引是0,还可以省略:
>>> L[:3]
['Michael', 'Sarah', 'Tracy']
# 既然Python支持L[-1]取倒数第一个元素,那么它同样支持倒数切片,试试:
>>> L[-2:]
['Bob', 'Jack']
>>> L[-2:-1]
['Bob']
# 所有数,每2个取一个:
>>> L[::2]
['Michael', 'Tracy', 'Jack']
# 只写[:]就可以原样复制一个list:
>>> L[:]
['Michael', 'Sarah', 'Tracy', 'Bob', 'Jack']
# tuple也可以用切片操作,只是操作的结果仍是tuple:
>>> (0, 1, 2, 3, 4, 5)[:3]
(0, 1, 2)
# 字符串'xxx'也可以看成是一种list,每个元素就是一个字符。因此,字符串也可以用切片操作,只是操作结果仍是字符串:
>>> 'ABCDEFG'[:3]
'ABC'
>>> 'ABCDEFG'[::2]
'ACEG'
迭代
# dict可以迭代
# 因为dict的存储不是按照list的方式顺序排列,所以,迭代出的结果顺序很可能不一样。
# 默认情况下,dict迭代的是key。如果要迭代value,可以用for value in d.values(),
# 如果要同时迭代key和value,可以用for k, v in d.items()。
>>> d = {'a': 1, 'b': 2, 'c': 3}
>>> for key in d:
... print(key)
...
a
c
b
# 由于字符串也是可迭代对象,因此,也可以作用于for循环:
>>> for ch in 'ABC':
... print(ch)
...
A
B
C
# Python内置的enumerate函数可以把一个list变成索引-元素对,
# 这样就可以在for循环中同时迭代索引和元素本身:
>>> for i, value in enumerate(['A', 'B', 'C']):
... print(i, value)
...
0 A
1 B
2 C
# 通过collections模块的Iterable类型判断一个对象是否可迭代对象
>>> from collections import Iterable
>>> isinstance('abc', Iterable) # str是否可迭代
True
>>> isinstance([1,2,3], Iterable) # list是否可迭代
True
>>> isinstance(123, Iterable) # 整数是否可迭代
False
列表生成式
# 如果要生成[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]
# 用列表生成式如下:
>>> [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']
# 列表生成式也可以使用两个变量来生成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']
生成器
在Python中,一边循环一边计算的机制,称为生成器。
## 只要把一个列表生成式的[]改成(),就创建了一个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
at 0x1022ef630>
# 如果要一个一个打印出来,可以通过next()函数获得generator的下一个返回值;
# 正确的方法是使用for循环,因为generator也是可迭代对象:
>>> g = (x * x for x in range(10))
>>> for n in g:
... print(n)
...
0
1
## 如果一个函数定义中包含yield关键字,那么这个函数就不再是一个普通函数,而是一个generator ##
# 定义一个generator,依次返回数字1,3,5
def odd():
print('step 1')
yield 1
print('step 2')
yield(3)
print('step 3')
yield(5)
# 把函数改成generator后,直接使用for循环来迭代
>>> for n in fib(6):
... print(n)
...
1
1
2
3
5
8
# 但是用for循环调用generator时,发现拿不到generator的return语句的返回值。
# 如果想要拿到返回值,必须捕获StopIteration错误,返回值包含在StopIteration的value中:
>>> g = fib(6)
>>> while True:
... try:
... x = next(g)
... print('g:', x)
... except StopIteration as e:
... print('Generator return value:', e.value)
... break
...
g: 1
g: 1
g: 2
g: 3
g: 5
g: 8
Generator return value: done
迭代器
凡是可作用于for循环的对象都是Iterable类型;
>>> from collections import Iterable
>>> isinstance([], Iterable)
True
凡是可作用于next()函数的对象都是Iterator类型,它们表示一个惰性计算的序列;
>>> from collections import Iterator
>>> isinstance((x for x in range(10)), Iterator)
True
集合数据类型如list、tuple、dict、set、str等是Iterable但不是Iterator,不过可以通过iter()函数获得一个Iterator对象。
>>> isinstance(iter([]), Iterator)
True