[問題] 請問有什麼辦法加快這個 for loop 嗎？

[CaptPlanet (ep)]

有list_a, list_b兩個list

list_a 有大約 70000 個 elements

list_b 大約 3 million 個 elements

程式大致如下：


res_li = []
for x in list_b:
    try:
        res_li.append(list_a.index(x))
    except:
        res_li.append("")


對 list_b 中的每一個 element

在 list_a 中找到一樣 element 把他的 index 加到新的 list 中

隨著 iteration 增加 速度變得越來越慢，

想請教各位為何會有這個現象以及有什麼方法加速這個 for loop 呢？

謝謝各位高手



--
※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 110.28.65.74
※ 文章網址: https://www.ptt.cc/bbs/Python/M.1519717995.A.701.html
→ uranusjr: 簡單的方法是把 list_a 的 index 用 dict cache 起來
→ uranusjr: 每個 b 裡的值都要在 a 從頭找一次當然會慢
→ CaptPlanet: 感恩 已解決
推 Yshuan: 做一個 dict([(v, i) for i, v in enumerate(list_a)])
→ Yshuan: 用空間換取時間這樣...
推 Yshuan: res_li = map(lambda k: new_dict.get(k, ""), list_b)
推 thefomalhaut: cython or numpy?
→ secondsee: 或可用set找出交集, 直接拿去list_a比對
→ secondsee: check = set(list_a) & set(list_b)
→ secondsee: res_li = [i for i,x in enumerate(a) if x in check]
推 vfgce: 樓上的解法有誤...也沒有比dict快.
推 vfgce: Y大解法ok,只是map回傳的是iterator,再轉成list似乎沒有
→ vfgce: 比直接用list comprehension快.
推 eggeggss: 2個list轉set後intersection再產生新的list