web-dev-qa-db-ja.com

海生まれのバイオリンプロット：バイオリンの色、軸の名前、凡例を変更する

最近matplotlibを発見するまで、私はseabornを使用してすべてのプロットを作成していました。与えられたhue変数を比較するためにバイオリンを分割できるという事実は、特に興味深いと思いました。

さて、私の主な問題は、バイオリンの色、軸の名前を変更するために何を変更する必要があるのか​​わからないことです）、または凡例を必要な場所に配置します。

これが私がseabornからたどった例です：

    import seaborn as sns
    tips = sns.load_dataset("tips")
    sns.set(style="ticks", palette="colorblind")
    g = sns.FacetGrid(tips, col="time", size=4, aspect=.75)
    g = g.map(sns.violinplot, "sex", "total_bill", "smoker", inner=None, linewidth=1, scale="area", split=True, width=0.75).despine(left=True).add_legend(title="smoker")
    g.savefig(os.path.join(options.output_dir, "figures", "violinplots.png"))

そして、これが出力図ですviolinplots.png：

enter image description here

私はむしろこのようなものが欲しいのですが：

enter image description here

要約する：

  • whiteとblueの色を使用する
  • 軸を置き換えるnames
  • 書く leftmost y axisのみ
  • 自分で作るlegend with the blue category only

前もって感謝します。どんな助けでも大歓迎です。

誰かが興味を持った場合のために、これが私が最終的に図を解決した方法です MrPedru22 ：

import seaborn as sns
tips = sns.load_dataset("tips")
sns.set(context="paper", palette="colorblind", style="ticks")
g = sns.FacetGrid(tips, col="time", sharey=False, size=4, aspect=.5)
g = g.map(seaborn.violinplot, "sex", "total_bill", "smoker", cut=0, inner=None, split=True, palette={"No": "#4477AA", "Yes": "white"}, saturation=1).despine(left=True)
# Set axis labels & ticks #
g.fig.get_axes()[0].set_xlabel("Lunch")
g.fig.get_axes()[1].set_xlabel("Dinner")
g.fig.get_axes()[0].set_xticklabels(["Male", "Female"])
g.fig.get_axes()[1].set_xticklabels(["Male", "Female"])
g.fig.get_axes()[0].set_ylabel("Total bill")
g.fig.get_axes()[0].set_yticks(range(0, 80, 10))
g.fig.get_axes()[1].set_yticks([])
g.fig.get_axes()[0].spines["left"].set_visible(True)
# Set legend #
handles, labels = g.fig.get_axes()[0].get_legend_handles_labels()
g.fig.get_axes()[0].legend([handles[1]], ["Non-smoker"], loc='upper left')
# Fixing titles #
g.fig.get_axes()[0].set_title("")
g.fig.get_axes()[1].set_title("")
g.plt.show()

enter image description here

pythonmatplotlibseaborn
13
ofornes

ここにいくつかの答えがあります：

import seaborn as sns
tips = sns.load_dataset("tips")
sns.set(style="ticks", palette="colorblind")
g = sns.FacetGrid(tips, col="time", size=4, aspect=.75)
g = g.map(sns.violinplot, "sex", "total_bill", "smoker", palette={"No": "b", "Yes": "w"}, inner=None, linewidth=1, scale="area", split=True, width=0.75).despine(left=True)
g.fig.get_axes()[0].legend(title= 'smoker',loc='upper left')
g.set_axis_labels('lunch','total bill')
sns.plt.show()
6
MrPedru22