2つのcsvの共通部分を新たなcsvに書き込む

Question

###前提・実現したいこと
【ソースコード訂正あり】
python3.5.1で２つのcsvの共通部分を新たなcsvに書き込みたいと思っています。
正規表現のserchを使って共通部分を取りだそうとしたのですができていません。

それぞれのcsvの中身は以下のようになっています。
●a.csv
melon,
banana,
apple,
︙
●b.csv
apple,
orange,
lemon,
︙

###発生している問題・エラーメッセージ

f3にnameのみ書かれており、共通部分を取り出せていない。

###該当のソースコード

import csv
import re

f1 = open('a.csv', 'rt')
reader1 = csv.reader(f1)
next(reader1)

f2 = open('b.csv', 'rt')
reader2 = csv.reader(f2)

f3 = open('c.csv','wt')
writer = csv.writer(f3, lineterminator='\n')
writer.writerow(['name'])

for columns1 in reader1: 
    for columns2 in reader2:
        name = re.search(columns1[0],columns2[0])
        if name:
            writer.writerow([name])

f1.close()
f2.close()
f3.close()

Answer 1

掲載されたコードはいくつか問題があります、おそらく期待した動作はしていないでしょう。

まず、csv.readerはイテレータ（ジェネレータ）なのでforの2重ループは2回目のループが上手くいかないはずです。
2重ループなどで繰り返し使う場合は、事前にlistでリストに変換してください。
詳しくはジェネレータを調べてみてください。

2点目、このケースならば正規表現を使うまでもありません、setで十分です。
多分、私が書くのであればこうします。

import csv

col1 = list(csv.reader(open('a.csv', 'rt')))
col2 = list(csv.reader(open('b.csv', 'rt')))

# 1行目はヘッダーなので捨てているのでしょうか？
col1 = [x[0].strip() for x in col1[1:]]
col2 = [x[0].strip() for x in col2[1:]]

writer = csv.writer(open('c.csv','wt'), lineterminator='\n')
writer.writerow(['name'])
for name in set(col1) & set(col2):
    writer.writerow([name])

私はコードは短くシンプルを目指しているので若干荒いですが、元コードはかなり丁寧にコードを書いていますね。
とても良いことだと思います。

できましたらwith構文などを使ったらもっと堅牢性が上がると思います。

Comments

[cir]

おっしゃる通り、1行目はヘッダーです。記載ミスでした。
2重ループにはリスト型の変換が必要なのですね。ジェネレータとsetについて後で調べておきます。

質問時はwith文で複数のファイルを開くことができないと勘違いしており、利用しなかったのですが、カンマ区切りで利用できるようなので、書き換えておきます。

pashango2さんのコードはシンプルですね。私も基本を抑えたら、シンプルなコードを目指したいと思います。回答ありがとうございます。理解が深まりました。

Answer 2

下記の一致の摘出部分がおかしいかと思います。

b.csvのリードでEOFまで読んでしまうので、先頭に
リードポインタを戻さないと2回目のループができない。
のが原因のようです。ですのでラフに直すと下記で動くかと思います。

import csv
import re

f1 = open("a.csv", "rt")
reader1 = csv.reader(f1)

f2 = open("b.csv", "rt")
reader2 = csv.reader(f2)

f3 = open("c.csv","wt")
writer = csv.writer(f3, lineterminator="\n")
writer.writerow(['name'])
list = []
for columns2 in reader2:
    list.append(columns2[0])
for columns1 in reader1: 
    for str in list:
        print(str)
        name = re.search(columns1[0],str)
        if name:
            writer.writerow([name.group(0)])

f1.close()
f2.close()
f3.close()

Comments

[cir]

確かにこの場合、クォーテーションマークは必要ありませんでした。ご指摘ありがとうございます。

しかし、クォーテーションマークをとってやってみてもf3にはnameしか表示されませんでした...

[nagaetty]

columns1[0],columns2[0]の中身を表示してみるのが近道かと思います！

[cir]

アドバイス通りcolumns1[0],columns2[0]の中身を表示して原因はわかったのですが、色々試しても解決方法がわからないままでした。
とても助かりました！回答ありがとうございます。