PandasのPivot_tableで計算されたvalueの数字の摘出について

Question

前提・実現したいこと

PandasのPivot_tableで顧客ID,商品ID,買い上げ点数の分析をしたいです

index='customerid', columns='itemid', values='numberofitems'

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

やりたい事は各顧客IDのtotal買い上げ点数が90以下だけを摘出したいのですが、pivot_tableで集計されたtotalをどのように処理したら良いのか分かりません。

勝手に想像しているのは、pivot_tableにより、買い上げ点数(numberofitems)がvalueになった事で列や行と認識されず、queryなどで90以下を摘出する、という事が出来ないのではないか、と思っております。

ご確認お願いいたします

Answer 1

こういうときは、小さなサンプルデータを自分で作って、自分で試行錯誤したり、質問文にサンプルデータを載せるなりしてください。

ピボットテーブルの結果を格納したデータフレームの変数そのものに対して変数 <= 90を実行すると、データフレームの各要素が90以下かどうかの真偽値(True/False)を要素とする新たなデータフレームを生成できます。あとは、それを利用して「あなたのやりたい抽出」(詳細が書かれていないので、以下ではやりません)を自分で行なえばいい。

補足: やりたいことが「顧客別totalが200以下の行を抽出し、顧客別totalだけの棒グラフを作りたい」だと判明したので、それに合わせて修正。

dfp[dfp['total'] <= 200]で顧客別totalの値が200以下のピボットデーブルの行がデータフレームとして得られます。顧客別totalの値だけが必要なら、dfp.loc[dfp['total'] <= 200, 'total']でシリーズが得られます。これらに対してplot.barすれば棒グラフが生成できます。

import pandas as pd
import io
import matplotlib.pyplot as plt # グラフ表示用に追加
import japanize_matplotlib # グラフで日本語表示用に追加

txt = """
顧客ID,アイテムID,数量
A,Item1,50
A,Item2,30
B,Item1,100
B,Item3,90
C,Item2,200
C,Item3,10
"""

# 日本語を含むデータフレームの出力を揃える
pd.set_option('display.unicode.east_asian_width', True)

# ファイルの代わりにtxtから読み込んでデータフレームを生成
df = pd.read_csv(io.StringIO(txt))

# ピボットテーブルを作成。欠損値を0に置換してから整数化
dfp = pd.pivot_table(df, index='顧客ID', columns='アイテムID', values='数量', aggfunc='sum', margins=True, margins_name='total').fillna(0).applymap(int)
print(dfp, end="\n\n")

# 顧客別のtotalの値が200以下の顧客だけ集めたデータフレームを生成
df2 = dfp[dfp['total'] <= 200]
print(df2, end="\n\n")
# df2.plot.bar()
# plt.show()

# 顧客別のtotalの値が200以下のtotalの値だけ集めたシリーズを生成してグラフを表示
s = dfp.loc[dfp['total'] <= 200, 'total']
print(s, end="\n\n")
s.plot.bar()
plt.show()

アイテムID  Item1  Item2  Item3  total
顧客ID
A              50     30      0     80
B             100      0     90    190
C               0    200     10    210
total         150    230    100    480

アイテムID  Item1  Item2  Item3  total
顧客ID
A              50     30      0     80
B             100      0     90    190

顧客ID
A     80
B    190
Name: total, dtype: int64

Comments

[Mario_]

ありがとうございました。すごく助かりました。

[Mario_]

Daregadaさん

申し訳ございません。
上記に関して勘違いしておりました。

Totalの値が90以下だけのデータフレームを作りたい場合はどうなりますでしょうか？？
色々記事読んだり試行錯誤しているのですが、どうしても進まなくて・・・。
Daregadaさんにいただいた例を使わせていただいた場合、

アイテムID Item1 Item2 Item3 total
顧客ID
A 50 30 0 80
B 100 0 90 190
C 0 200 10 210
total 150 230 100 480

顧客のトータル購入数が200以下の顧客IDを抜き出したデータフレームを作りたい、という事です、
この場合だと、AとBさんだけが残る様な関数を教えていただけないでしょうか？？

何卒よろしくお願いいたします。

[Mario_]

また、可能であればグラフ(棒グラフ)の作り方も教えていただきたいです。

X軸を顧客IDで、Y軸をTotalの数量だけにする方法が分からず、私の元データでdf.plot.bar()で単純にやるとデータが重すぎてエラーになります。なので、totalだけの数量でグラフを作りたいです。

何卒ご教示いただけると大変助かります。
よろしくお願いいたします

[Mario_]

ありがとうございます！！

なんとか自分で出来る様にがんばります！！