本篇将介绍Python的列表生成式,更多内容请参考:Python列表生成式
列表生成式即List Comprehensions,是Python内置的非常简单却强大的可以用来创建list的生成式。
举个例子,要生成list
可以用
- [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
- list(range(1, 11))
- >>>list(range[1, 11])
- [1, 2, 3, 4, 5, 6, 7, 8, 9, 10,]
但是要生成
怎么做?方法一是循环:
- [1×1, 2×2, 3×3, 4×4,..., 10×10]
- >>>L = []
- >>>for x in range(1, 11):
- L.append(x * x)
- >>>L
- [1, 4 ,9 ,16, ..., 100]
但是循环太繁琐,而列表生成式则可以用一行语句代替循环生成上面的list:
- >>> [x * x
- for x in range(1, 11)][1, 4, 16, 25, 36, 49, 64, 81, 100]
写列表生成式时,把要生成的元素
放在前面,后面跟
- x * x
循环,就可以把list创建出来,十分有用,多写几次,很快就可以熟悉这种语法。 for循环后面还可以加上if判断,这样我们就可以筛选出仅偶数的平方:
- for
- >>> [x * x
- for x in range(1, 11) if x % 2 == 0][4, 16, 64, 100]
还可以使用两层循环,还可以生成全排列:
- >>>[m+n for m in 'ABC' for n in 'XYZ']
- ['AX', 'AY', 'AZ', 'BX', 'BY', 'BZ', 'CX', 'CY', 'CZ']
三层和三层以上的循环就很少使用了。
运用列表生成式,可以写出非常简洁的代码。例如,列出当前目录下的所有文件和目录名,可以通过一行代码实现:
- >>> import os#导入os模块 >>> [d
- for d in os.listdir('.')]#os.listdir可以列出文件和目录
循环其实可以同时使用两个甚至多个变量,比如
- for
的
- dict
可以同时迭代key和value:
- items()
- >>>d = {'X':'A', 'y':'B', 'z':'C'}
- >>>for k, v in d.items():
- print(k, '=', 'v')
- y = B
- x = A
- z = C
因此,列表生成式也可以使用两个变量来生成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']
1、使用for循环的迭代不仅可以迭代普通的list,还可以迭代dict。
假设有如下的dict:
d = {'Adam':95, 'Lisa':85, 'Bart':59}
完全可以通过一个复杂的列表生成式把它变成一个html表格:
- tds = ['<tr><td>%s</td><td>%s</td></tr>'%(name, score) for name, score in d.items()]
- print('<table>')
- print('<tr><th>Name</th><th>Score</th></tr>')
- print '\n'.join(tds)
- print('</table>')
注意:字符串可以通过%进行格式化,用指定的参数替代%s。字符串的join()方法可以把一个list拼接成一个字符串。
2、在生成的表格中,对于没有及格的同学,请把分数标记为红色。
- d = {'Adam':95, "Lisa":85, "Bart":59}
- def generate_tr(name, score):
- if score < 60:
- return '<tr><td>%s</td><td style="color:red">%s</td></tr>' %s(name, score)
- return '<tr><td>Name</td><td>Score</td></tr>'%(name, score)
- tds = [generate_tr(name, score) for name, score in d.items()]
- print('<table border = "1">')
- print('\n'.join(tds))
- print('</table>')
3、条件过滤
列表生成式的for循环后面还可以加上if判断,例如:
- >>> [x * x
- for x in range(1, 11)][1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
如果我们只想要偶数的平方,不改动range()的情况下,可以加上if来筛选:
- >>> [x * x
- for x in range(1, 11) if x % 2 == 0][4, 16, 64, 100]
有了if条件,只有if判断为True的时候,才把循环的当前元素添加到列表中。
编写一个函数,它接收一个list,然后把list中的所有字符串变成大写后返回,非字符串元素将被忽略。
- [x.upper()
- for
- x
- in
- L
- if
- isinstance
- (x,
- str
- )]
来源: http://www.cnblogs.com/miqi1992/p/7784449.html