pandasのpivot_tableで要素ごとにカウント

Question

pandasのpivot_tableを使って、要素数ごとにカウントをしたいのですが、
要素がカウントされずに困っております。

▼試したこと

import pandas as pd

# データフレームを作る
df = pd.DataFrame({"name":["tarou", "tarou", "tarou", "tarou", "hanako", "hanako", "hanako", "jun", "jun"], "food":["orange", "apple", "banana", "orange", "banana", "orange", "apple", "orange", "banana"]})

# 要素ごとにカウント
df.pivot_table(index="name", columns= "food", aggfunc='count')

上記の結果、要素ごとのカウント数が出てこず、このような結果が返ってきます。

本来は、以下のような結果が返ってくることを想定しています。

pivot_tableのコードで何かご指摘があればご教示頂けますでしょうか。

Answer 1

.pivot_table()では、内部でGroupByを使っており、indexとcolumnsにデータフレームのすべての列を使い切っていると「値の列」がないため'count'を指定するとそのように空白になります。

df.pivot_table(index=index, columns=columns, aggfunc=aggfunc)
# ↑これは
# ↓これにほぼ等しい
df.groupby([index, columns]).agg(aggfunc).unstack(columns)

In [10]: df.groupby(['name', 'food']).count()
Out[10]:
Empty DataFrame
Columns: []
Index: [(hanako, apple), (hanako, banana), (hanako, orange), (jun, banana), (jun, orange), (tarou, apple), (tarou, banana), (tarou, orange)]

このような場合に数を数えたい場合は'size'（lenでも可能）を指定します。

In [14]: df.groupby(['name', 'food']).size().unstack('food', fill_value=0)
Out[14]:
food    apple  banana  orange
name
hanako      1       1       1
jun         0       1       1
tarou       1       1       2

In [15]: df.pivot_table(index='name', columns='food', aggfunc='size', fill_value=0)
Out[15]:
food    apple  banana  orange
name
hanako      1       1       1
jun         0       1       1
tarou       1       1       2

なお、pd.crosstab()や、pandasバージョン1.1.0以降であればvalue_counts()メソッドを使う方法もあります（いずれの方法でも、中身でやっていることはほとんど同じです）。

In [16]: pd.crosstab(df['name'], df['food'])
Out[16]:
food    apple  banana  orange
name
hanako      1       1       1
jun         0       1       1
tarou       1       1       2

In [17]: df.value_counts().unstack('food', fill_value=0)
Out[17]:
food    apple  banana  orange
name
hanako      1       1       1
jun         0       1       1
tarou       1       1       2

Comments

[color_8]

kiraraさん
ご丁寧な回答ありがとうございました！

pivot_tableの内部ではGroupByを行っているのですね。
実は最初GroupByで実装しようとしたのですが、分からなかったためpivotでチャレンジしていました。

非常に勉強になりました！
ありがとうございます。