Pandasでqueryを使って条件抽出する方法

Question

クラスタリングをして、その後、満足度が4または5のものだけを抽出してヒートマップで可視化したいです。ヒートマップのプログラムはこのプログラムの下に書きます。
満足度は「1~5」、性別は「男性」か「女性」、個人/グループは「個人」か「グループ」、旅行の目的は「学生・ビジネス・航空関係・旅行・その他」で回答されているアンケート結果のcsvファイルを読み込んでいます。

_df = df_cluster1.groupby(['満足度','性別', '個人/グループ', '旅行の目的']).median().query('"満足度" == 4 or "満足度" == 5'),


エラー
TypeError: argument of type 'int' is not iterable

エラーが解決されず、どこを直せば良いのか教えていただけると幸いです。

query('"満足度" == 4 or "満足度" == 5')

の部分を自分で書きました。

Answer 1

_df = df_cluster1.groupby(['満足度','性別', '個人/グループ', '旅行の目的']).median().query('"満足度" == 4 or "満足度" == 5')

を

_df = df_cluster1.groupby(['満足度','性別', '個人/グループ', '旅行の目的']).median().query('満足度 == 4 or 満足度 == 5')

に変更しましょう。

Comments

[egpro]

解決しました！ありがとうございました！