python(4) list生成式

title: python(4) list生成式
tag: python

---

要生成一个列表list，如果这个list有一定的规律，可以用循环生成。python的内置函数range()可以直接生成list
如下

>>> range(1, 11)
[1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

在matlab里面，如果要生成这样一个矩阵，首先定义一个空矩阵，然后通过for循环不断往里push值，在python里面也可以这样操作，但是代码写起来太麻烦了，比如说我们要生成一个list
[1x1, 2x2, 3x3, ..., 10x10]
用for循环代码如下

L=[]
for x in range(1,11)
    L.append(x*x)

其实在python里面，不用for循环，一行列表生成式就可以达到上面的目的

一层循环

>>> [x*x for x in range(1,11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

把要生成的元素x * x放到前面，后面跟for循环，就可以把list创建出来

两层循环

>>> [m + n for m in 'ABC' for n in 'XYZ']
['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']

由结果可以看出:两个字符串相加是字符串

两个变量的循环

其实在之前字典里，利用dict.iteritem()就可以同时循环键和键值

>>> d = {'x': 'A', 'y': 'B', 'z': 'C' }
>>> for k, v in d.iteritems():
...     print k, '=', v
... 
y = B
x = A
z = C

现在列表生成器也可以用两个变量来生成list

 d = {'x': 'A', 'y': 'B', 'z': 'C' }
[k+'='+v for k,v in d.iteriterms()]
['y=B', 'x=A', 'z=C']

最后把一个list中所有的字符串变成小写：

>>> L = ['Hello', 'World', 'IBM', 'Apple']
>>> [s.lower() for s in L]
['hello', 'world', 'ibm', 'apple']

s指代列表中的所有字符串，s.lower()方法是把大写字母变成小写，是字符串的方法，如果有整数在里面，就会报错
三层以上循环就很少用了