python列表推导式-轻量级循环

列表推导式是利用其它列表创建新列表。它的工作方式类似于for循环

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

列表由range（10）中每个x的平方组成，太容易了？如果想打印那些能被3整除的平方数呢？可以使用模除运算符x%3.当数字可以被3整除时返回0.可以通过增加一个if部分添加到列表推导中。

>>> [x*x for x in range(10) if x %3 ==0]
[0, 9, 36, 81]

也可以增加更多的for语句的部分：

>>> [(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)]

作为对比，下面的代码使用两个for循环创建了相同的列表：

result = []
for x in range(3):
   for y in range(3):
        result.append((x,y))
print result

[(0, 0), (0, 1), (0, 2), (1, 0), (1, 1), (1, 2), (2, 0), (2, 1), (2, 2)]

也可以和if子句联合使用，像以前一样

>>> boys = ['sunquan','weiqing','chenkun']
>>> girls = ['sunli','weizifu','chenzihan']
>>> [b + '+' + g for b in boys for g in girls if b[0]==g[0]]
['sunquan+sunli', 'weiqing+weizifu', 'chenkun+chenzihan']

注意：使用普通的圆括号而不是方括号。否则不会得到‘元祖推导式’

上面男孩女孩的例子，其实效率不高。因为它会检查每个可能的配对。python有很多解决这个问题的方法：

boys = ['sunquan','weiqing','chenkun']
girls = ['sunli','weizifu','chenzihan']
letterGirls = {}
for girl in girls:
    letterGirls.setdefault(girl[0],[]).append(girl)
print [b + '+' + g for b in boys for g in letterGirls[b[0]]]
['sunquan+sunli', 'weiqing+weizifu', 'chenkun+chenzihan']

这个程序建造了一个letterGirl的字典，其中每一项都把单字母作为键，以女孩名字组成的列表作为值。在字典建立后，列表推导式循环整个男孩集合，并且查找那些和当前男孩首字母相同的女孩集合。这样列表推导式就不用尝试所有的男孩女孩的组合，检查首字母是否匹配。