07-python学习笔记-列表生成式

2017.6.4
参考资料

---

列表生成式-廖雪峰的官方网站

---

可以使用List Comprehensions来创建list，它是Python内置的，非常简单，非常强大。

例子1

要生成list[1,2,3,4,5,6]，使用list(range(1,7))

>>> list(range(1,7))
[1, 2, 3, 4, 5, 6]

例子2

要生成[1x1, 2x2, 3x3, …, 10x10]，怎么做？

当我们使用循环的时候

>>> L=[]
>>> for x in list(range(1,11)):
...     L.append(x*x)
...
>>>
>>> L
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

当我们使用列表生成式时

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

for循环后面还可以加上if判断，这样我们就可以筛选出符合自己条件的数字，如果我们想筛选出偶数的平方

>>> [x*x for x in list(range(1,10)) if x%2==0]
[4, 16, 36, 64]

使用两层循环，生成全排列

>>> [x+y for x in '123' for y in 'abc']
['1a', '1b', '1c', '2a', '2b', '2c', '3a', '3b', '3c']

复习使用06篇笔记中的迭代，使用两个变量来生成list

这里在做的时候遇到一个小问题

>>> dict={'a':1,'b':2,'c':3}
>>> [k + '=' + v for k,v in dict.items()]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <listcomp>
TypeError: Can't convert 'int' object to str implicitly

value不能是整形
更改value成string类型，重新做

>>> dict={'a':'A','b':'B','c':'C'}
>>> [k + '=' + v for k,v in dict.items()]
['b=B', 'c=C', 'a=A']

练习

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

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

但是，当list中既包含字符串，又包含整数，由于非字符串类型没有lower()方法，所以列表生成式会报错

>>> L = ['Hello', 'World', 'IBM', 'Apple',13]
>>> [s.lower() for s in L]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "<stdin>", line 1, in <listcomp>
AttributeError: 'int' object has no attribute 'lower'

下面，修改列表生成式，通过添加if语句保证列表生成式能正确地执行，提示，内建的isinstance函数可以判断一个变量是不是字符串：

>>> [s.lower() for s in L if isinstance(s,str)==True]
['hello', 'world', 'ibm', 'apple']