Dostosowywanie adnotacji z FacetGrid Seaborn

Question

Próbuję dostosować niektóre figury z modułem Seaborn w Pythonie, ale nie miałem szczęścia tworzenia niestandardowych etykiet i adnotacji. Mam niektóre kodu, który generuje następujący rysunek:

plot = sns.FacetGrid(data = data, col = 'bot', margin_titles = True).set_titles('Human', 'Bot') 
bins = np.linspace(0, 2000, 15) 
plot = plot.map(plt.hist, 'friends_count', color = 'black', lw = 0, bins = bins) 
plot.set_axis_labels('Number Following', 'Count') 
sns.despine(left = True, bottom = True) 

chciałbym zrobić dwie rzeczy: 1. Wymień etykiety czynników, na przykład domyślne "bot = 0,0", z tekstem znaczącym i 2. narysuj linie pionowe o średniej następującej po każdej kategorii.

Oto samowystarczalny przykład:

import pandas as pd 
import seaborn as sns 
import matplotlib.pyplot as plt 

fake = pd.DataFrame({'val': [1, 2, 2, 3, 3, 2, 1, 1, 2, 3], 'group': [0, 0, 0, 0, 0, 1, 1, 1, 1, 1]}) 
plot = sns.FacetGrid(data = fake, col = 'group', margin_titles = True).set_titles('zero', 'one') 
plot = plot.map(plt.hist, 'val', color = 'black', lw = 0) 
sns.despine(left = True, bottom = True) 

Ktoś wie jak dostosować FacetGrids?

Answer 1

Kilka rzeczy o set_titles.

pierwsze tytuły domyślne są rysowane w sposób FacetGrid.map, więc jeśli chcesz zmienić tytuły, trzeba zadzwonić set_titlespo kreślenia, albo zostaną one nadpisane.

Po drugie, jeśli spojrzysz na docstring dla metody, nie bierze ona tylko arbitralnej listy tytułów. Jest to sposób, aby zmienić sposób tytuł wydanego przy użyciu nazwy zmiennej i wartość kolumna:

template : string 
    Template for all titles with the formatting keys {col_var} and 
    {col_name} (if using a `col` faceting variable) and/or {row_var} 
    and {row_name} (if using a `row` faceting variable). 

więc najprostszym sposobem, aby mieć „znaczący tekst” jest użycie znaczących danych w dataframe. Weźmy ten przykład z losowych danych:

df = pd.DataFrame({'val': np.random.randn(100), 
        'group': np.repeat([0, 1], 50)}) 

Jeśli chcesz „grupa” będzie zero i one, należy po prostu zmienić tę kolumnę, albo zrobić nowe:

df["group"] = df["group"].map({0: "zero", 1; "one"}) 

Wtedy powiedzieć don „t chcą mieć nazwę zmiennej w tytule właściwy sposób wykorzystywać FacetGrid.set_titles byłoby

g = sns.FacetGrid(data=df, col='group') 
g.map(plt.hist, 'val', color='black', lw=0) 
g.set_titles('{col_name}') 

Jeśli nie chcesz zmienić dane masz kreślenia, następnie trzeba będzie ustawić atrybuty na matplotlib osiach bezpośrednio, coś takiego:

for ax, title in zip(g.axes.flat, ['zero', 'one']): 
    ax.set_title(title) 

pamiętać, że jest mniej korzystne do powyższej metody, ponieważ trzeba być bardzo ostrożnym w upewnianiu się, że kolejność na liście jest poprawna i że nie ulegnie ona zmianie, natomiast uzyskanie informacji z samej ramki danych będzie znacznie bardziej niezawodne.

Aby narysować średnią, musisz utworzyć małą funkcję, którą można przekazać do FacetGrid.map. Jest multiple examples, jak to zrobić w samouczku.W tym przypadku jest to dość proste:

def vertical_mean_line(x, **kwargs): 
    plt.axvline(x.mean(), **kwargs) 

Wtedy wszystko, co potrzebne jest do ponownego działki:

g = sns.FacetGrid(data=df, col='group') 
g.map(plt.hist, 'val', color='black', lw=0) 
g.map(vertical_mean_line, 'val') 
g.set_titles('{col_name}')