csvファイルを読み込んで出力したい

Question

import csv
import termcolor

input_name = input('こんにちは！私はRobokoです。あなたの名前は何ですか？')
input_restaurant_name = input(f'{input_name}さん。どこのレストランが好きですか？')
count = 0

with open('test.csv', 'w+', newline='') as csv_file:
    fieldnames = ['Name', 'Count']
    writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
    writer.writeheader()
    reader = csv.DictReader(csv_file)
    #わからない
    writer.writerow({'Name': input_restaurant_name.capitalize(), 'Count': count})

上のように、csvファイルで、フィールドを作って、inputされたものを書き込むようにしたのですが、2回目以降の書き込みで、input_restaurant_nameに同じものがあったらcountを+1したいです。

Name,Count
Apple_restaueant,1

Comments

[meg_]

test.csvのサンプルを提示してください。

[meg_]

まず、「with open('test.csv', 'w+', newline='') as csv_file:」を実行するとtest.csvの内容は空になります。なのでカウントできません。

Answer 1

test.csvの内容を読み取って、入力されたレストランがが同じものがあればカウントしてtest.csvに書き込む例です。

import csv

input_name = input('こんにちは！私はRobokoです。あなたの名前は何ですか？')
input_restaurant_name = input(f'{input_name}さん。どこのレストランが好きですか？')

with open('test.csv', 'r', newline='') as csv_file:
    reader = csv.DictReader(csv_file)
    l = [row for row in reader]
    for i in range(len(l)):
        if l[i]['Name'] == input_restaurant_name:
            l[i]['Count'] = int(l[i]['Count']) + 1
            break
        if i == len(l) -1:
            l.append({'Name':input_restaurant_name, 'Count':1})

with open('test.csv', 'w', newline='') as csv_file:
    fieldnames = ['Name', 'Count']
    writer = csv.DictWriter(csv_file, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerows(l)