转自廖雪峰的讲义
切片
对于list、tuple和字符串等类型的数据都可以使用切片取出指定的元素。
L = ['Michael', 'Sarah', 'Tracy', 'Bob', 'Jack']
取前3个元素:L[0:3]或者L[:3]
L[0:3]表示,从索引0开始取,直到索引3为止,但不包括索引3。
L[-1]取倒数第一个元素。
前10个数,每两个取一个:L[:10:2]
迭代:
通过for循环来遍历可迭代的对象(如list, tuple, dict, 字符串等),这种遍历我们称为迭代。
迭代dict:
d = {'a': 1, 'b': 2, 'c': 3}
for key in d:
... print key
d = {'x': 'A', 'y': 'B', 'z': 'C' }
for k, v in d.iteritems():
... print k, '=', v
判断一个对象是否可迭代:
from collections import Iterable
isinstance('abc', Iterable)
列表生成器:
将要生成的元素写在前面,后面跟for循环,就可以使用列表生成器创建list。
[x * x for x in range(1, 11)]
[m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']
L = ['Java' , 'C' , 'Swift' , 'Python' , 123]
print [s.lower() if isinstance(s , str) else s for s in L ]
生成器:
生成器(generator)可以按照设定好的算法一边迭代,一边计算值。
将列表生成式的[]改成(),就创建了一个generator:
g = (x * x for x in range(10))
可以使用for循环对generator中的元素进行迭代:
g = (x * x for x in range(10))
for n in g:
... print n
如果推算的算法比较复杂,可以用函数或者generator来实现。例如,生成斐波拉契数列:
def fib(max):
n, a, b = 0, 0, 1
while n < max:
print b
a, b = b, a + b
n = n + 1
使用关键字yield,可以将函数转化为generator:
def fib(max):
n, a, b = 0, 0, 1
while n < max:
yield b
a, b = b, a + b
n = n + 1
注意:变成generator的函数,在每次调用next()的时候执行,遇到yield语句返回,再次执行时从上次返回的yield语句处继续执行。