如下：

links.extend(link for link in get_links(html) if re.match(link_regex, link))
# 上面的語句分開寫,如下
# 但下面需要使用links.append,如果依然使用links.extend會出現錯誤的結果
for link in get_links(html):
  if re.match(link_regex, link):
     links.append(link)

讓我們用ipython做個實驗:

由上圖可以看出,for in if 連擊推導式隱式的將link由str轉變為了list，而分開寫是不會幫我們轉換的，所以需要使用append把整個str加入到list.

補充：python的for i in list 陷阱

d = [1,2,3,4,5]
for i in d:
  d.remove(i)
print d

結果是 :

[2, 4]

這個陷阱比較隱蔽,我再用白盒子測試的時候,發現了這個bug,大致原因就是remove d[0]之后,i 自動變成了 d[1], 但是 d已經變成了 d[2,3,4,5],所以i=d[1]就悲催的指向了3,跳過了2!

看來python還是要注意了解底層實現啊!

解決辦法是

d = [1,2,3,4,5]
for i in d[:]:
  d.remove(i)
print d

就可以實現邊遍歷邊刪除了

以上為個人經驗，希望能給大家一個參考，也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方，望不吝賜教。