pandasでのpivotテーブル作成時の総数の追加方法について

Question

pandasでpivotテーブルを作成し要素をカウントして、その総数をテーブルに追加したいです。

以下の例のdataframeからpd.pivotを用いて都市の出現数をカウントしました。
そのカウントの総数を加えたいのですがエラーとなり実現できておりません。
カウントの総数を加える方法をご教示頂けたらありがたいです。

<例のdataframe:df>
Date 国名 都市
10/1 日本 東京
10/1 韓国 ソウル
10/1 日本 大阪
10/1 韓国 ソウル
10/1 日本 大阪
10/1 韓国 蔚山
10/2 韓国 蔚山
10/2 日本 大阪
10/2 日本 大阪
10/2 日本 東京
10/2 韓国 ソウル

＜作成できたpivotイメージ＞
|国名|都市　|10/1|10/2
|日本|東京　|1 |1
|　　|大阪　|2 |2
|韓国|ソウル|2 |1
|　　|蔚山　|1 |1

＜実現したいpivotイメージ＞
|国名|都市　|10/1|10/2|総数
|日本|東京　|1 |1　　|2
|　　|大阪　|2 |2　　|4
|韓国|ソウル|2 |1　　|3
|　　|蔚山　|1 |1　　|2
|総数　　　 |6 |5　　|11

＜総数を加える前のエラーとならなかったコード＞

df1 = pd.pivot_table(df , index=['国名','都市'] , columns='Date' , aggfunc='size')

＜総数を加えようとしてエラーとなったコード＞

df1 = pd.pivot_table(df , index=['国名','都市'] , columns='Date' , aggfunc='size' , margins=True)

エラーはKeyErrorと表示されました。

Answer 1

aggfunc=lenを指定すればよいです。
参考：pandasのピボットテーブルでカテゴリ毎の統計量などを算出

なお、len()を使うとカテゴリごとの出現回数が算出できる

import pandas as pd
from io import StringIO
pd.set_option('display.unicode.east_asian_width', True)

s = """Date,国名,都市
10/1,日本,東京
10/1,韓国,ソウル
10/1,日本,大阪
10/1,韓国,ソウル
10/1,日本,大阪
10/1,韓国,蔚山
10/2,韓国,蔚山
10/2,日本,大阪
10/2,日本,大阪
10/2,日本,東京
10/2,韓国,ソウル"""
df = pd.read_csv(StringIO(s))

df1 = pd.pivot_table(df , index=['国名','都市'] , columns='Date' , aggfunc=len , margins=True)
print(df1)
"""
Date         10/1  10/2  All
国名 都市
日本 大阪       2     2    4
     東京       1     1    2
韓国 ソウル     2     1    3
     蔚山       1     1    2
All             6     5   11
"""

Comments

[zumzum]

ご教示いただきましてありがとうございます。
試してできました。感謝です。参考サイトもありがとうございます。