pandas によるアンケート集計について

Question

pythonの質問です。

pandas を利用してアンケートの集計をしたいと考えています。
下記の様な表があったときに、既出の回答が来たときに該当した行のカウントを＋1、
初めての回答が来たときに新たな行に要素を追加するプログラムを作成したいです。
作成したコードではえらーが発生してしまい、上手くいきません。エラーの発生原因を教えて頂ければ幸いです。以上、よろしくおねがい致します。

【データフレーム】
restaurant名　　count
cocosu　　　　　 2
saizeria 1

【作成してみたコード】
restaurant = str(input("レストラン名>"))

df = pd.DataFrame([[None,0]], columns=['restaurant名', 'count'],encoding="shift-jis")

if restaurant in df.restaurant名:
df.loc[df['restaurant名'] == restaurant,"count"] += 1

print(df)

else:
df2 = pd.DataFrame([[restaurant,1]], columns=['restaurant名', 'count'] ,encoding="shift-jis")
df = df.append(df2)

【発生したエラー】
count
restaurant名
NaN 0
レストラン名>gasuto
Traceback (most recent call last):
File "/Users/pythonProject4/firstcontact.py", line 27, in <module>
if restaurant in df.restaurant名:
File "/Users/pythonProject4/venv/lib/python3.8/site-packages/pandas/core/generic.py", line 5465, in getattr
return object.getattribute(self, name)
AttributeError: 'DataFrame' object has no attribute 'restaurant名'

Comments

[meg_]

コードは「コードの挿入｣で記入してください。

[ppaul]

Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: __init__() got an unexpected keyword argument 'encoding'
となります。
このコードは動かないのではありませんか？

Answer 1

本当のソースコードが提示されていないので推測ですが、
'restaurant名'がdfのインデクスになっているようです。

if restaurant in df.restaurant名:
の前に、

df.reset_index(, inplace=True)

を入れれば、

AttributeError: 'DataFrame' object has no attribute 'restaurant名'

は出なくなる可能性が高いです。

その後が思ったように動くかどうかは別問題です。

Comments

[huchamon]

ppaulさんの書かれていた通りで、インデクスとされていた様でした。他にもいくつか不具合はあったもののお陰様でなんとか解決致しました。ありがとうございました