Rotular eixos no Seaborn Barplot


125

Estou tentando usar minhas próprias etiquetas para um gráfico de barra Seaborn com o seguinte código:

import pandas as pd
import seaborn as sns

fake = pd.DataFrame({'cat': ['red', 'green', 'blue'], 'val': [1, 2, 3]})
fig = sns.barplot(x = 'val', y = 'cat', 
                  data = fake, 
                  color = 'black')
fig.set_axis_labels('Colors', 'Values')

insira a descrição da imagem aqui

No entanto, recebo um erro que:

AttributeError: 'AxesSubplot' object has no attribute 'set_axis_labels'

O que da?

Respostas:


235

O barplot de Seaborn retorna um objeto de eixo (não uma figura). Isso significa que você pode fazer o seguinte:

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

fake = pd.DataFrame({'cat': ['red', 'green', 'blue'], 'val': [1, 2, 3]})
ax = sns.barplot(x = 'val', y = 'cat', 
              data = fake, 
              color = 'black')
ax.set(xlabel='common xlabel', ylabel='common ylabel')
plt.show()

2
seabornnão tem sua própria maneira de configurá-los - sem envolver matplotlib?
javadba

Portanto, a regra geral é FacetGrid/ qualquer coisa que facetas retorne um objeto de figura e todo o resto retorne um objeto de eixo?
alexpghayes

27

Pode-se evitar o AttributeErrorcausado pelo set_axis_labels()método usando o matplotlib.pyplot.xlabele matplotlib.pyplot.ylabel.

matplotlib.pyplot.xlabeldefine o rótulo do eixo x enquanto matplotlib.pyplot.ylabeldefine o rótulo do eixo y do eixo atual.

Código da solução:

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

fake = pd.DataFrame({'cat': ['red', 'green', 'blue'], 'val': [1, 2, 3]})
fig = sns.barplot(x = 'val', y = 'cat', data = fake, color = 'black')
plt.xlabel("Colors")
plt.ylabel("Values")
plt.title("Colors vs Values") # You can comment this line out if you don't need title
plt.show(fig)

Figura de saída:

insira a descrição da imagem aqui


13

Você também pode definir o título do seu gráfico adicionando o parâmetro title da seguinte forma

ax.set(xlabel='common xlabel', ylabel='common ylabel', title='some title')
Ao utilizar nosso site, você reconhece que leu e compreendeu nossa Política de Cookies e nossa Política de Privacidade.
Licensed under cc by-sa 3.0 with attribution required.