Etiquetar ejes en Seaborn Barplot

125

Estoy tratando de usar mis propias etiquetas para un diagrama de barras de Seaborn con el siguiente 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')

ingrese la descripción de la imagen aquí

Sin embargo, aparece un error que:

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

¿Lo que da?

Erin Shellman
fuente

Respuestas:

235

El diagrama de barras de Seaborn devuelve un objeto-eje (no una figura). Esto significa que puede hacer lo siguiente:

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()
Sascha
fuente
2
seabornno tiene su propia manera de configurarlos, sin involucrar matplotlib?
javadba
Entonces, ¿la regla general es FacetGrid/ cualquier cosa que facete devuelva un objeto figura y todo lo demás devuelva un objeto eje?
alexpghayes
27

Se puede evitar el método AttributeErrorprovocado set_axis_labels()por el uso de matplotlib.pyplot.xlabely matplotlib.pyplot.ylabel.

matplotlib.pyplot.xlabelestablece la etiqueta del eje x mientras que matplotlib.pyplot.ylabelestablece la etiqueta del eje y del eje actual.

Código de solución:

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 salida:

ingrese la descripción de la imagen aquí

Steffi Keran Rani J
fuente
13

También puede establecer el título de su gráfico agregando el parámetro de título de la siguiente manera

ax.set(xlabel='common xlabel', ylabel='common ylabel', title='some title')
John R
fuente