python pandas.pivot_tableにキャプションを付けたい

前提・実現したいこと

pythonでwebスクレイピングした情報をもとに、ピボットテーブルを作成し、
自身のサイトにそのテーブルを載せることを目的としたプログラムを作成しています。

その前段として、リスト形式のデータからピボットテーブルを作り、
htmlを作成することまでできたのですが、そのテーブルにキャプションを付けられず困っています。

アドバイス頂戴できると幸いです。

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

性別女男
果物
みかん NaN 4.0
りんご 6.0 3.0
レモン 5.0 NaN

というテーブルの上に「果物一覧」というキャプション(タイトル)が表示されてほしいのですが、
何も表示されません

該当のソースコード

import pandas as pd

a=["りんご","みかん","レモン","りんご"]
b=["男","男","女","女"]
c=[3,4,5,6]
df= pd.DataFrame({"果物": a, "性別" :b,"個数":c})
print(df.pivot_table(values = "個数", index = "果物", columns = "性別"))
df_pivot=df.pivot_table(values = "個数", index = "果物", columns = "性別")

df_pivot.style.set_caption("果物一覧") #←この行で作成したピボットテーブルに\キャプション(表タイトル)がつくという認識なのですが。。。
print(df_pivot)
html=df_pivot.to_html()

text_file = open(r"C:\MypythonScript\index.html","w")
text_file.write(html)　#←この行で作成したHTMLを開いてもやはりキャプションはついていません
text_file.close()

試したこと

.style.set_captionを用いてキャプションを付けられるという認識なのですが、うまくいかず詰まっております。
エラーは特に出ていません。

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

行動規範の内容に同意します

回答1件

ベストアンサー

style.set_captionメソッドで設定されたキャプションを含めたHTMLを生成するには、従来のto_htmlではなく、キャプションを設定済みのスタイルに対してstyle.renderメソッドを使います。

Styler Object and HTML

The above output looks very similar to the standard DataFrame HTML representation. But the HTML here has already attached some CSS classes to each cell, even if we haven’t yet created any styles. We can view these by calling the .render() method, which returns the raw HTML as string, which is useful for further processing or adding to a file - read on in More about CSS and HTML.

キャプションを設定したスタイルは、style.set_captionの返り値として得られるので、これを保存しておかないといけません。
その他のスタイル(背景色など)を追加設定する際にも、「今までの設定が行なわれたスタイル」に対して新たなスタイルを設定するメソッドを実行し(チェーン可能)、返り値を保存する必要があります。

Python
1import pandas as pd
2
3a = ["りんご", "みかん", "レモン", "りんご"]
4b = ["男", "男", "女", "女"]
5c = [3, 4, 5, 6]
6
7df = pd.DataFrame({"果物": a, "性別": b, "個数": c})
8print(df.pivot_table(values="個数", index="果物", columns="性別"))
9df_pivot = df.pivot_table(values="個数", index="果物", columns="性別")
10
11style = df_pivot.style.set_caption("果物一覧")
12html = style.render()
13print(html)
14
15with open("sample_style.html", "w", encoding="utf-8") as text_file:
16    text_file.write(html)