[Python]CSVファイルへの辞書形式での書き込みについて質問です。

Question

初めまして。閲覧の程有難う御座います。
現在以下のような使用のアプリケーションをオンラインスクールの課題で制作しております。

#課題の仕様

• まず、こちらが名前を聞く

• そのあとに、おすすめのレストランの名前を聞く

• そのレストラン名を変数に入れ、カウントをし、（レストラン名がかぶった場合はカウントを1増やす）CSVファイルに書き込む。

その際の、CSVに書き込むコードをPythonで書いていたのですが、どうも詰まってしまっています。理想は

NAME, COUNT
ガスト, 2
バーミヤン, 1

と書き込めるようなpythonコードを作りたいので御座います。

現在、書き込みを行う段階で詰まってしまっているので御指南の程をお願い致します。

count = 1
#csvモジュールはインポート済み
with open("ranking.csv", "w", newline="") as r:
    
    fieldnames = ["NAME", "COUNT"]
    writer = csv.DictWriter(r, fieldnames=fieldnames)
    writer.writeheader()
    writer.writerows({"NAME": Restaurant_name, "COUNT": count})

こちらを実行すると

Traceback (most recent call last):
  File "C:/Users/nhs90438/PycharmProjects/Python_Programming/recommend.py", line 29, in <module>
    writer.writerows({"NAME": Restaurant_name, "COUNT": count})
  File "C:\Users\nhs90438\Anaconda3\lib\csv.py", line 158, in writerows
    return self.writer.writerows(map(self._dict_to_list, rowdicts))
  File "C:\Users\nhs90438\Anaconda3\lib\csv.py", line 148, in _dict_to_list
    wrong_fields = rowdict.keys() - self.fieldnames
AttributeError: 'str' object has no attribute 'keys'

このようなエラーが吐かれます。string型のオブジェクトはkeyの属性を持っていないよ！ってのはわかるのですが、やはり素人だからなのか「・・・はあ。」となって対処法が本当にわかりません。いったい何を直せばよいのでしょうか・・・ご回答お待ちしております。

実行環境：

• python3.7

• Pycharm

• Windows10

Comments

[hayataka2049]

変数 Restaurant_nameと countに代入されているのはどのようなオブジェクトでしょうか？

[Apex-Riflex]

ご回答の程有難うございます。大変申し訳ないのですが、自分自身まだオブジェクトという概念についてこれといった知識が深めれていなくて、ご質問者様の意図が読み取れないでいます。ご質問者が仰りたいのは
どのような値がそれらの変数に代入されているかというご質問でよろしいでしょうか？そちらでしたら、
Restaurant_nameにはユーザーからのstr型の入力値が入っています。そしてcountにはこちらからint型の1をとりあえず代入しております。ご期待に添えたつもりですが間違っていたら申し訳ございません。その際はよろしければ詳しく指摘していただくと今後の糧にもなりますので、お願い致します。

Answer 1

こういうときはまず仕様を確認しましょう！！！

今回はcsvモジュールのページの、

csv.DictWriter | csv --- CSV ファイルの読み書き — Python 3.8.1 ドキュメント
writer オブジェクト | csv --- CSV ファイルの読み書き — Python 3.8.1 ドキュメント

などが参考になります。

特に、writer オブジェクトの説明の

row には、 Writer オブジェクトの場合には文字列か数値のイテラブルを指定し、 DictWriter オブジェクトの場合はフィールド名をキーとして対応する文字列か数値を格納した辞書オブジェクトを指定します(数値は str() で変換されます)。

---

csvwriter.writerow(row)

Write the row parameter to the writer's file object, formatted according to the current dialect. Return the return value of the call to the write method of the underlying file object.

---

csvwriter.writerows(rows)
rows 引数 (上で解説した row オブジェクトのイテラブル) の全ての要素を現在の表現形式に基づいて書式化し、writer のファイルオブジェクトに書き込みます。

などの記述が役に立ちます。