PTT數位生活區 / Python

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

看板Python作者 (ep)時間7年前 (2018/02/27 15:53), 編輯推噓6(6推 0噓 8→)
留言14則, 7人參與, 6年前最新討論串1/2 (看更多)
有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
→
02/27 16:03, 7年前 , 1F
簡單的方法是把 list_a 的 index 用 dict cache 起來
02/27 16:03, 1F
→
02/27 16:03, 7年前 , 2F
每個 b 裡的值都要在 a 從頭找一次當然會慢
02/27 16:03, 2F
→
02/27 16:21, 7年前 , 3F
感恩 已解決
02/27 16:21, 3F
推
02/27 16:51, 7年前 , 4F
做一個 dict([(v, i) for i, v in enumerate(list_a)])
02/27 16:51, 4F
→
02/27 16:51, 7年前 , 5F
用空間換取時間這樣...
02/27 16:51, 5F
推
02/27 16:54, 7年前 , 6F
res_li = map(lambda k: new_dict.get(k, ""), list_b)
02/27 16:54, 6F
推
02/28 00:53, 7年前 , 7F
cython or numpy?
02/28 00:53, 7F
→
02/28 10:36, 7年前 , 8F
或可用set找出交集, 直接拿去list_a比對
02/28 10:36, 8F
→
02/28 10:41, 7年前 , 9F
check = set(list_a) & set(list_b)
02/28 10:41, 9F
→
02/28 10:43, 7年前 , 10F
res_li = [i for i,x in enumerate(a) if x in check]
02/28 10:43, 10F
推
02/28 11:58, 7年前 , 11F
樓上的解法有誤...也沒有比dict快.
02/28 11:58, 11F
推
02/28 12:03, 7年前 , 12F
Y大解法ok,只是map回傳的是iterator,再轉成list似乎沒有
02/28 12:03, 12F
→
02/28 12:04, 7年前 , 13F
比直接用list comprehension快.
02/28 12:04, 13F
推
10/27 16:42, 6年前 , 14F
2個list轉set後intersection再產生新的list
10/27 16:42, 14F
‣ 更多 CaptPlanet 的文章
文章代碼(AID): #1QbGvhS1 (Python)
Python 近期熱門文章
更多 近期熱門文章 >>
PTT數位生活區 即時熱門文章
更多 即時熱門文章 >>
‣ 更多 CaptPlanet 的文章
文章代碼(AID): #1QbGvhS1 (Python)