python的列表產生式可以把要產生的元素x*x放在前面,後面跟上一個for迴圈即可,這是python特有的列表產生式,可以以非常簡潔的代碼產生list。
ex:>>> [x * x for x in range(1, 11)]
[1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
range(1,100,2)可以產生[1,3,5,7…….97,99]利用這種模式可以產生更靈活的列表。
使用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 拼接成一個字串。
把列印出來的結果儲存為一個html檔案,就可以在瀏覽器中看到效果了:
<table border="1"><tr><th>Name</th><th>Score</th><tr><tr><td>Lisa</td><td>85</td></tr><tr><td>Adam</td><td>95</td></tr><tr><td>Bart</td><td>59</td></tr></table>
列表產生式的 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, 36, 64, 100]下面是一個將list集合中所有字串變成大寫,非字串被忽略def toUppers(L): return [ x.upper() for x in L if isinstance(x,str)]print toUppers(['Hello', 'world', 101])
for迴圈可以嵌套,因此,在列表產生式中,也可以用多層 for 迴圈來產生列表。
對於字串 ‘ABC’ 和 ‘123’,可以使用兩層迴圈,產生全排列:
>>> [m + n for m in 'ABC' for n in '123']['A1', 'A2', 'A3', 'B1', 'B2', 'B3', 'C1', 'C2', 'C3']
翻譯成迴圈代碼就像下面這樣:
L = []for m in 'ABC': for n in '123': L.append(m + n)