生成器快速创建数组:
1、一维数组
>>> print([i *2 for i in range(10)]) #range(n)表示0到(n-1)的整数
[0, 2, 4, 6, 8, 10, 12, 14, 16, 18]
>>> print([i * i for i in range(10) if (i % 3) == 0])
[0, 9, 36, 81]
>>> print([(x, y) for x in range(3) for y in range(3)])
[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]
2、二维数组
>>> a = [[3] * (i + 1) for i in range(3)]
>>> a
[[3], [3, 3], [3, 3, 3]]
>>> a[2]
[3, 3, 3]
我们知道列表乘以某数时表示重复该片段数次,但运用时容易出现乘法问题,如:
>>> b = [[1, 2, 3]] *3
>>> b[1][1] = 100
>>> b
[[1, 100, 3], [1, 100, 3], [1, 100, 3]]
可以发现在最终结果中不仅只更改了b[1][1],还影响了其他元素。
上述过程可以更改为:
>>> b = [[1, 2, 3] for i in range(3)]
>>> b[1][1] = 100
>>> b
[[1, 2, 3], [1, 100, 3], [1, 2, 3]]
经典案例:
实现斐波那契数列
>>>def fib(num):
n, a, b = 0, 0, 1
while n < num:
print(b)
a,b = b, a+b
n += 1
return 'done'
>>>print(5)
1
1
2
3
5
Done!