seaborn.facetgridでの文字追加

Question

seabornのfacetgridで複数のグラフを作成するときに、
グラフに文字を書き込むこむ方法を教えて欲しいです。
具体的には、複数のヒストグラムをあるパラメーターで層別して表示したいのですが、
それぞれのヒストグラムのAveとSigmaの数字も表示させたいです。
(AveとSigmaは事前に計算するとして)

以下はseabornのfacetgridの公式サイトです。
https://seaborn.pydata.org/generated/seaborn.FacetGrid.html

Comments

[arch_]

最終的なアウトプットのイメージが欲しいです。

[satopooooon]

コメントありがとうございます。イメージ画像を追加しました。ヒストグラムのAveとsigmaをグラフ上に記載したいです。

Answer 1

seabornはmatplotlibのラッパですから、各gridに対応するaxisを取得してtext()を呼べば良いです。ドキュメントを読んだらfacet_axisで取れると書いてありました。

seaborn.FacetGrid — seaborn 0.8.1 documentation

import numpy as np
import seaborn as sns
import matplotlib.pyplot as plt

bins = np.arange(0, 65, 5)
tips = sns.load_dataset("tips")
g = sns.FacetGrid(tips, col="time",  row="smoker")
g = g.map(plt.hist, "total_bill", bins=bins, color="r")

for i in range(2):
    for j in range(2):
        ax = g.facet_axis(i, j)
        ax.text(40,20,"hoge")

plt.show()

あとは頑張ってください。

追記

それぞれのgridに対応するデータの取得で困るかもしれないので書いておくと、g.facet_data()で場所とデータの情報が得られるようです。これに対してave, sigmaを計算しておいて辞書にでも格納しておけば、なんとかなると思います。

Comments

[satopooooon]

ありがとうございます。できました！