Aリストの０番目の要素とリストの０番目の要素が一致するものを残したい

Question

例えば、[[100,23]]と[100]は互いの０番目の要素が一致しているため、
[[100,23]]を残す、という具合です。

a=[[100,23],[101,34],[102,35],[103,22],[104,55]・・・[2000000,21]]
b=[100,104,107・・・2000000]

ans = []
for i in range(len(a)):
      for j in range(len(b)):
            if a[i][0] == b[j]:
                 ans.append(a[i])

とコードを書いて実行したところ、計算が一向に終わりません。
aリスト・bリストともに、
2000000近くデータがあるからだと思いましたが、これ以外の方法が思いつきませんでした。
より高速にこのコードを書きなおすにはどうしたらいいですか？

Comments

[LouiS0616]

取得したansはその後どのように利用するのですか。

[k0908]

ansをfor文で回して、ansにどのようなものが含まれているか確認するために使います

Answer 1

ansをfor文で回して、ansにどのようなものが含まれているか確認するために使います

それならば、まず遅延評価を利用するのが第一かと思います。
実行速度は速くなりませんが、処理が分散するので実用はし易いはずです。

gen_ans = filter(lambda e: e[0] in b, a)
for ans in gen_ans:
    ...

また、帰属のチェックは単にin演算子を使えば良いです。
bの型もlistよりsetの方が扱いやすいでしょう。

Comments

[k0908]

ありがとうございます。遅延評価というワードを初めて聞いたため勉強になりました。質問なのですが、”bの型もlistよりsetの方が扱いやすい”と書かれた理由は何でしょうか？

[LouiS0616]

リストは要素の重複を許し、同時に順序も保証します。
しかし、bの各要素の値は重複しないはずですし、順序にも意味がないはずです。

数列ではなく単に集合としてbを用いるなら、そのための型であるsetを使った方が効率的です。

[k0908]

なるほど、そういうことでしたか。ありがとうございます！！