2つのリストのループ

Question

下記テーブル1の「番号」の値とテーブル2の「番号」の値を比較し、
テーブル1に存在して、テーブル2に存在しないものを出力したい場合、どのような方法が適切でしょうか？

テーブル1
番号　ID
0001　A
0002　B
0004　D
0006　D
0007　A
0009　E

テーブル2
番号　担当
0001 Yamada
0003　Hibino
0004　Ogawa
0006　Yamada
0007　Hanano
0010　Mizuno

ソートしてあるのでforループをテーブル1で回し、その中でforループでテーブル2を回す方法しか思いつかないのですが、処理速度などを考慮し、より良い方法があれば教えていただけると幸いです。
よろしくお願いいたします。

Comments

[meg_]

> ソートしてあるのでforループをテーブル1で回し、その中でforループでテーブル2を回す方法しか思いつかないのですが
そのコードを掲載していただけませんか？

[quickquip]

テーブル と呼んでいるものがいったいなんなのか、質問からは分からないのでコードを書きましょう。

Answer 1

リスト内包表記と all を使って抽出するサンプルコードです。

from csv import reader
from io import StringIO

# テーブル1
s1 = """番号,ID
0001,A
0002,B
0004,D
0006,D
0007,A
0009,E"""
# テーブル2
s2 = """番号,担当
0001,Yamada
0003,Hibino
0004,Ogawa
0006,Yamada
0007,Hanano
0010,Mizuno"""

lst1 = [row for row in reader(StringIO(s1))][1:]
lst2 = [row for row in reader(StringIO(s2))][1:]
dif = [ls1 for ls1 in lst1 if all(ls1[0] != ls2[0] for ls2 in lst2)]
print(dif)  # [['0002', 'B'], ['0009', 'E']]

Answer 2

番号がリストなどに入っているなら、set型(集合型)を使うのが高速です。

>>> num1 = ['0001', '0002', '0004', '0006', '0007', '0009']
>>> num2 = ['0001', '0003', '0004', '0006', '0007', '0010']
>>>
>>> print(set(num1)- set(num2))
{'0002', '0009'}

Comments

[abide]

ありがとうございます。set型使ったことがなかったので、大変勉強になりました。

Answer 3

pandasを使うと以下のような感じで簡潔に抽出できます。

import pandas as pd
from io import StringIO

# テーブル1
s1 = """番号,ID
0001,A
0002,B
0004,D
0006,D
0007,A
0009,E"""
# テーブル2
s2 = """番号,担当
0001,Yamada
0003,Hibino
0004,Ogawa
0006,Yamada
0007,Hanano
0010,Mizuno"""
df1 = pd.read_csv(StringIO(s1), dtype=str)
df2 = pd.read_csv(StringIO(s2), dtype=str)

df = df1[ ~df1['番号'].isin(df2['番号'])] # 2に存在しないものを抽出
print(df)
#     番号 ID
#1  0002  B
#5  0009  E

Comments

[abide]

ありがとうございます！ループで行うよりとても速くて大変勉強になりました。