[Python]csvで一部のデータをカウントアップして更新したい

Question

前提・実現したいこと

コードを実行するとおすすめのレストラン名を聞かれます。
おすすめのレストラン名の質問に対しての回答はrestaurant.csvにて、
"Name"キーと"Count"キーにて保存されます。
"Name"キーには質問の回答のレストラン名が入り、
"Count"キーには質問に対しての回答で何回そのレストラン名が出てきたかをカウントします。

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

初めて名前が出てきたレストランはしっかりとcsvに記録がされるのですが、
二回目以降名前が出てきたレストラン名に関しては上手く"Count"キーがカウントアップされません、
何故でしょうか。

該当のソースコード

import csv
import os
 
 
print("#####################################################")
print("どこのレストランが好きですか？")
print("#####################################################")
restaurant = input()
 
# csvファイルが無かったら作成+書き加える→else、在ったら書き加える→if。
if os.path.exists("restaurant.csv"):
    with open("restaurant.csv", "r+", newline='') as res_csv:
        fieldnames = ["Name", "Count"]
        writer = csv.DictWriter(res_csv, fieldnames=fieldnames)
        # addはレストランの名前があるか否かのトリガー
        restaurant_is_exist = False
        # res_csvをfor文でListにできるように変換
        new_csv = csv.reader(res_csv)
        for row in new_csv:
            print(row) # ここのrowはlist型
            if (restaurant in row) == True:
                print(int(row[1]) + 1)
                row[1] = int(row[1]) + 1
                print("Restaurant is Exist")
                restaurant_is_exist = True
        if restaurant_is_exist == False:
            writer.writerow({"Name": restaurant, "Count": 1})
            print("Restaurant is not Exist")
    print("If")
else:
    with open("restaurant.csv", "w", newline='') as res_csv:
        fieldnames = ["Name", "Count"]
        writer = csv.DictWriter(res_csv, fieldnames=fieldnames)
        writer.writeheader()
        writer.writerow({"Name": restaurant, "Count": 1})
    print("Else")

試したこと

コード中の、

print(int(row[1]) + 1)

で、実際にカウントアップ出来るのかを確認したりしてみたのですが、
カウントアップは出来ているようでした、しかしそれがrestaurant.csvには反映されていないようです。

補足情報

restaurant.csvの中身です（画像添付出来なかったのでリンクで失礼します）、
これは二回動作させて一回目をhoge、二回目をhogehogeにした結果です。
画像

Answer 1

初めて名前が出てきたレストランはしっかりとcsvに記録がされるのですが、

二回目以降名前が出てきたレストラン名に関しては上手く"Count"キーがカウントアップされません、
何故でしょうか。

if (restaurant in row) == True:
print(int(row[1]) + 1)
row[1] = int(row[1]) + 1
print("Restaurant is Exist")
restaurant_is_exist = True

変数rowはあくまでもcsvから読み取った結果であって、
書き換えれば自動的にcsvファイルが更新されるわけではありません。

手動で当該行を書き直してやる必要があります。

---

プログラム冒頭でcsvファイルを全て読み込み、
データを更新した後、プログラム終了直前に再度書き出すのが楽なように思います。

この場合ファイルが既に存在するかに依って書き出しモードを変える必要もありません。

Comments

[kei-kappa564]

回答を頂きありがとうございます。
データベース上でやっているわけでは無いので、
csv上に上書きしなければならなく出来なかったとは私の勉強不足でした。

重ね重ね質問になってしまい申し訳ないのですが2つほどお聞きしたいことがあります。

①手動で当該行を書き直しすにはどの様な方法があるのでしょうか。appendなどを用いれば出来るのでしょうか。

➁「プログラム冒頭でcsvファイルを全て読み込み、
データを更新した後、プログラム終了直前に再度書き出すのが楽なように思います。」
とは、csvを全て読み込んだ後、レストラン名を追加 or カウントアップを行ってから全文csvに上書きということでしょうか。

お手数をおかけしますが、
お暇な時で良いのでよろしくお願いいたします。

[LouiS0616]

①開いているファイルをそのまま編集する方法もありますが、大抵コードが難解になるので完全に作り直す方が楽です。

②概ね、はい。上書きというか同名ファイルの作り直しですね。

Answer 2

カウントアップされないのは、前回までの結果を読み込んでいないのが原因のため、以下のようにしてみてはどうでしょうか。

サンプルコード

import csv
import os

data = {}

# ファイルが存在すれば、読み込む。
if os.path.exists('data.csv'):
    with open('data.csv', encoding='utf-8') as f:
        data = {name: int(count) for name, count in list(csv.reader(f))}

print('#####################################################')
print('どこのレストランが好きですか？')
print('#####################################################')
name = input()

# 存在する場合はカウントアップ、そうでない場合はキー追加
if name in data:
    data[name] += 1
else:
    data[name] = 1

# ファイルに書き込む。
with open('data.csv', 'w', encoding='utf-8') as f:
    writer = csv.writer(f)
    for name, count in data.items():
        writer.writerow([name, count])

実行例

ロイヤルホスト,2
マクドナルド,1
ジョナサン,2

入力

#####################################################
どこのレストランが好きですか？
#####################################################
はなまるうどん

はなまるうどん,1
ロイヤルホスト,2
マクドナルド,1
ジョナサン,2

追記

# list(reader) は [row for row in reader] と同じ
lst = list(reader)
print(lst)
# [['はなまるうどん', '1'], ['ロイヤルホスト', '3'], ['マクドナルド', '1'], ['ジョナサン', '2']]

for name, count in lst:
    print(name, count)
# はなまるうどん 1
# ロイヤルホスト 3
# マクドナルド 1
# ジョナサン 2

lst を dict の内包表現 を使って dict にしています。

dic = {i: i * 2 for i in range(5)}
print(dic)  # {0: 0, 1: 2, 2: 4, 3: 6, 4: 8}

Comments

[kei-kappa564]

とても分かりやすい問題解決に繋がるコードを丁寧に書いていただきありがとうございます。

重ね重ね質問になってしまい申し訳ないのですがPython初心者でして一つ質問があります、

data = {name: int(count) for name, count in list(csv.reader(f))}

ここの部分は何をしているのでしょうか。
dict形式で作っていると思ったのですが...。

[tiitoi]

dict の内包記法になります。
[['はなまるうどん', '1'], ['ロイヤルホスト', '3'], ['マクドナルド', '1'], ['ジョナサン', '2']]
というリストを
{'はなまるうどん': 1, 'ロイヤルホスト': 3, 'マクドナルド': 1, 'ジョナサン': 2}
にしています。

追記したので、参考にしてください。

[kei-kappa564]

詳しく追記して頂きありがとうございます。
どの様に動いているのかある程度理解できるようになりました。

三度失礼なのですが、1つ問題が発生しまして、
先ほどの教えて頂いたコードを1回目実行するのは問題なく動いたのですが、
2回目の実行の時に以下の様なエラーが出てしまいました。

Traceback (most recent call last):
File "C:/Users/〇〇〇/PycharmProjects/python_programming/practice01-kai.py", line 9, in <module>
data = {name: int(count) for name, count in list(csv.reader(f))}
File "C:/Users/〇〇〇/PycharmProjects/python_programming/practice01-kai.py", line 9, in <dictcomp>
data = {name: int(count) for name, count in list(csv.reader(f))}
ValueError: not enough values to unpack (expected 2, got 0)
※〇は伏字

これは値が足りていないエラーなのは理解できましたが、
どの様に解決すればよろしいのでしょうか。

[tiitoi]

csv の各列は `レストラン名,数` となっていることを想定しています。
編集されたりして空行が入ってしまったのかもしれません。
```
data = {row[0]: int(row[1]) for row in list(csv.reader(f)) if row}
```
に書き換えてみてはどうでしょうか。if row で row が空文字の場合は無視するようにします。

[kei-kappa564]

書き換えたら出来ました、ありがとうございます。
他にも下記の一行に、
```
with open('data.csv', 'w', encoding='utf-8') as f:
```
newline=''を付け加えても改行が取り払われて出来ました。
```
with open('data.csv', 'w', encoding='utf-8', newline='') as f:
```

色々とお手数をおかけしまして申し訳ありません、
ありがとうございました。