PythonでCSVファイルの共通部分を出力する方法

Question

3つのCSVファイルから共通部分を抜き出し、エディア上に出力したいです。

filea: Brazil,Japan,China,Korea
fileb: US,Brazil,France,Spain
filec: Portugal,Italy,Brazil

出力したいもの Brazil

Answer 1

1. ファイルをそのまま読み込む
2. カンマでsplitして文字列のリストにする (フィールドにカンマが含まれない前提)
3. setに放り込んで集合オブジェクトにする
4. 上記の手順をファイル数の分だけ繰り返す
5. 集合全体の共通部分を求める
6. 出力

---

参考：

• 組み込み型 - str — Python 3.7.4 ドキュメント
• 組み込み型 - set — Python 3.7.4 ドキュメント

Comments

[takayukichi]

読み込みしかできてなくて、splitしてリストにする以降はどう書けますか？

[LouiS0616]

こういうときの為に『参考』を貼っているのです。
https://docs.python.org/ja/3/library/stdtypes.html#str.split

Answer 2

ファイルから読み込み、以下のようにreduceで処理すればよいかと思います。

from functools import reduce

files = [['B','J','C','K'],['U','B','F','S'],['P','I','B']]
s =reduce(lambda a,b: set(a) & set(b), files)
print(s) # {'B'}

Comments

[LouiS0616]

set.intersectionの引数は可変長なので set.intersection(*map(set, files)) で済んだりします。

[can110]

ありゃ、なるほど。簡単ですね。

Answer 3

csvの各行毎に共通項目を取り出すようにしてみます。
filea.csv

Brazil,Japan,China,Korea
"a,b",ab,ac

fileb.csv

US,Brazil,France,Spain
"a,b",ab,ax

filec.csv

Portugal,Italy,Brazil
"a,b",ab

a.py

import csv

csvs = [
    "filea.csv", "fileb.csv", "filec.csv"
]

common_rows = []
for csv_file in csvs:
    with open(csv_file) as f:
        reader = csv.reader(f)
        if len(common_rows) == 0:
            for row in reader:
                common_rows.append(set(row))
        else:
            idx = 0
            for row in reader:
                common_rows[idx] = common_rows[idx] & set(row)
                idx += 1

print(common_rows)

実行例