質問をすることでしか得られない、回答やアドバイスがある。

15分調べてもわからないことは、質問しよう!

ただいまの
回答率

90.38%

  • Python 3.x

    10729questions

    Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

  • 正規表現

    964questions

    正規表現とは特定の文字列によるパターンマッチングを行う際に用いられる宣言型プログラミングです。

  • CSV

    922questions

    CSV(Comma-Separated Values)はコンマで区切られた明白なテキスト値のリストです。もしくは、そのフォーマットでひとつ以上のリストを含むファイルを指します。

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

解決済

回答 2

投稿 編集

  • 評価
  • クリップ 0
  • VIEW 907

cir

score 12

前提・実現したいこと

【ソースコード訂正あり】
python3.5.1で2つの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()
  • 気になる質問をクリップする

    クリップした質問は、後からいつでもマイページで確認できます。

    またクリップした質問に回答があった際、通知やメールを受け取ることができます。

    クリップを取り消します

  • 良い質問の評価を上げる

    以下のような質問は評価を上げましょう

    • 質問内容が明確
    • 自分も答えを知りたい
    • 質問者以外のユーザにも役立つ

    評価が高い質問は、TOPページの「注目」タブのフィードに表示されやすくなります。

    質問の評価を上げたことを取り消します

  • 評価を下げられる数の上限に達しました

    評価を下げることができません

    • 1日5回まで評価を下げられます
    • 1日に1ユーザに対して2回まで評価を下げられます

    質問の評価を下げる

    teratailでは下記のような質問を「具体的に困っていることがない質問」、「サイトポリシーに違反する質問」と定義し、推奨していません。

    • プログラミングに関係のない質問
    • やってほしいことだけを記載した丸投げの質問
    • 問題・課題が含まれていない質問
    • 意図的に内容が抹消された質問
    • 広告と受け取られるような投稿

    評価が下がると、TOPページの「アクティブ」「注目」タブのフィードに表示されにくくなります。

    質問の評価を下げたことを取り消します

    この機能は開放されていません

    評価を下げる条件を満たしてません

    評価を下げる理由を選択してください

    詳細な説明はこちら

    上記に当てはまらず、質問内容が明確になっていない質問には「情報の追加・修正依頼」機能からコメントをしてください。

    質問の評価を下げる機能の利用条件

    この機能を利用するためには、以下の事項を行う必要があります。

回答 2

checkベストアンサー

+1

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

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()

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2016/11/14 13:47

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

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

    キャンセル

  • 2016/11/14 15:13

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

    キャンセル

  • 2016/11/16 13:50

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

    キャンセル

+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構文などを使ったらもっと堅牢性が上がると思います。

投稿

編集

  • 回答の評価を上げる

    以下のような回答は評価を上げましょう

    • 正しい回答
    • わかりやすい回答
    • ためになる回答

    評価が高い回答ほどページの上位に表示されます。

  • 回答の評価を下げる

    下記のような回答は推奨されていません。

    • 間違っている回答
    • 質問の回答になっていない投稿
    • スパムや攻撃的な表現を用いた投稿

    評価を下げる際はその理由を明確に伝え、適切な回答に修正してもらいましょう。

  • 2017/05/30 11:29

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

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

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

    キャンセル

15分調べてもわからないことは、teratailで質問しよう!

  • ただいまの回答率 90.38%
  • 質問をまとめることで、思考を整理して素早く解決
  • テンプレート機能で、簡単に質問をまとめられる

同じタグがついた質問を見る

  • Python 3.x

    10729questions

    Python 3はPythonプログラミング言語の最新バージョンであり、2008年12月3日にリリースされました。

  • 正規表現

    964questions

    正規表現とは特定の文字列によるパターンマッチングを行う際に用いられる宣言型プログラミングです。

  • CSV

    922questions

    CSV(Comma-Separated Values)はコンマで区切られた明白なテキスト値のリストです。もしくは、そのフォーマットでひとつ以上のリストを含むファイルを指します。