Pythonでcsvファイルの特定の行の値を更新したい

Question

前提・実現したいこと

pythonでのcsv操作についての質問です。
inputで「好きな果物の名前」を取得し、'Name'カラムに果物の名前を、'Count'カラムにその果物の登場回数を
csvファイルに書き加えていくという処理を書こうとしています。
新たな果物を新規の行に書き込む処理はわかるのですが、「既存の果物の名前を取得したときにその行の'Count'を一つ増やすという処理」がどうしてもわかりません。

Name,Count
Apple,1
Banana,1
Peach,1
Orange,1

例えばinput()で'Orange'を取得した場合、

Name,Count
Apple,1
Banana,1
Peach,1
Orange,2

このように更新したいです。
今は標準ライブラリだけでやろうとしているのですが、サードパーティーのライブラリ使用などの選択肢も含めてどなたがご教授頂きたいです。
よろしくお願いいたします。

追記：

fruit_name = input()
fruit_name_title = furit_name.title()

if os.path.exists('fruits_ranking.csv'):
    with open('fruits_ranking.csv', 'r+', newline='') as f:
        reader = csv.DictReader(f)
        for i in reader:
            if i['Name'] == fruit_name_title:
                '......'

else:
    with open('fruits_ranking.csv', 'w', newline='') as f:
    fieldnames = ['Name', 'Count']
    writer = csv.DictWriter, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerow({'Name': fruit_name_title, 'Count': 1})

現状の説明が足りなくて申し訳ありません。
自分で考えた流れとしては、
・果物の名前を受け取る
・'fruits_ranking.csv'のファイルが存在するか調べる
・無ければファイルを新規で作り{'Name': fruit_name_title, 'Count': 1}を書き込む
・ファイルが存在する場合は各行をforループで回し、if i['Name'] == fruit_name_title:に引っかかった行に対して['Count']の更新はできないだろうかと考えていました。

一連の条件分岐に関しては一応機能するものを書けたのですが、csvの更新に関しては一度テキストファイルの形に変換してから部分的に上書きし、再度csvに反映させるなど考えてはみたのですが、実際にコードに起こせるだけの引き出しもなく行き詰ってしまいました。何かしらアドバイス頂けるとありがたいです。

補足情報（FW/ツールのバージョンなど）

Windows10
Python3
Pycharm

Comments

[meg_]

> 「既存の果物の名前を取得したときにその行の'Count'を一つ増やすという処理」がどうしてもわかりません。

"既存の果物の名前"かどうかの判定は出来ていますか？

Answer 1

inputが行われる毎にcsvファイルの一部を更新するのは非常に難しいです。csvファイルの1行の長さは決っていないので、ある部分だけを書き換えることが出来ないためです。

そこで、csvファイルを更新するような処理を行う場合は、プログラムの最初にcsvファイルを読み込んでおき、必要に応じてそれを変更しプログラムの最後でcsvファイルに書き込むのが普通です。

csvファイルを読み書きする方法としては、以下の方法があります。
0. テキストファイルとして読み込み、","で区切って要素を取り出し、テキストファイルとして書き出す。数字は自分で変換する。データ形式も自分で決める。
0. csvモジュールのreaderやwriterを使って読み書きする。数字は自分で変換する。データ形式は通常はリストのリストを使う。
0. numpyモジュールのloadtxt、savetxtを使う。今回のように、文字列と数字の両方が含まれる場合は、genfromtxtで読み込む。数字は自動的に変換される。データ形式はStructured arrayとなる。
0. pandasモジュールのread_csvやto_csvを使う。数字は自動的に変換される。データ形式はデータフレームとなる。

pythonの基礎を理解したいならテキストファイル方式をお勧めします。
処理の簡単さを優先するならpandasをお勧めします。覚えるべきことは多いですが将来役に立つことも多いです。

Comments

[kome1125]

やはりcsvファイルの性質上、私が望んでいるような処理をストレートに行うことは難しいのですね。

Answer 2

Pandas使うのが楽じゃないかな。

import pandas as pd
import io

txt = """
Name,Count
Apple,1
Banana,1
Peach,1
Orange,1
"""

df = pd.read_csv(io.StringIO(txt), index_col='Name')
# df = pd.read_csv("fruits.csv", index_col='Name')
print(df)

while True:
    fruit_name = input("好きな果物は? ")
    if len(fruit_name) == 0:
        break
    if fruit_name in df.index.values:
        df.loc[fruit_name] += 1
    else:
        df.loc[fruit_name] = 1
    print(df)

df.to_csv("fruits.csv")

Comments

[kome1125]

pandasについて調べてみようと思います、ありがとうございます。