Python开发学习笔记(11) - 列表生成式

列表生成式

列表生成式

列表生成式是Python中特有的写法，把要生成的元素放到前面，后面跟for循环，就可以用非常简洁的代码生成复杂的list了。

1.普通的列表生成可以直接用range(1,101)

print range(1,101)
>>[1,2,3,4....100]

2.生成复杂的列表， 如需要生成 [1*1, 2*2, 3*3 … 10*10]的列表。

　　1. 比较繁琐的方法是循环：

L = []
for x in range(1,11):
L.append(x * x)
print L
# 结果：[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

　　2. 比较简单的方法是列表生成式：

print [x * x for x in range(1, 11)]
# 结果：[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

# 如需要生成列表 [1x2, 3x4, 5x6, 7x8, ..., 99x100]

print [x*(x+1) for x in range(1,100,2)]

# range(1,100,2)可以生成list [1, 3, 5, 7, 9,...]

复杂表达式

使用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.iteritems()]
print '<table>'
print '<tr><th>Name</th><th>Score</th><tr>'
print '\n'.join(tds)
print '</table>'

# 注：字符串可以通过 % 进行格式化，用指定的参数替代 %s。字符串的join()方法可以把一个 list 拼接成一个字符串。

效果如下：



如果需要把不及格的分数标记成红色：

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>' % (name, score)
return '<tr><td>%s</td><td>%s</td></tr>' % (name, score)

tds = [generate_tr(name, score) for name, score in d.iteritems()]
print '<table border="1">'
print '<tr><th>Name</th><th>Score</th><tr>'
print '\n'.join(tds)
print '</table>'

条件过滤

列表生成式的 for 循环后面可以加上 if 判断，只有 if 判断为 True 的时候，才把循环的当前元素添加到列表中。

>>> [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, 36, 64, 100]