Мітки осей на Seaborn Barplot

Я намагаюся використовувати власні мітки для барплота Seaborn із наступним кодом:

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')

Однак я отримую помилку, що:

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

Що дає?

Барплот Seaborn повертає вісь-предмет (а не фігуру). Це означає, що ви можете зробити наступне:

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()

Можна уникнути AttributeErrorвикликаного set_axis_labels()методом, використовуючи matplotlib.pyplot.xlabelі matplotlib.pyplot.ylabel.

matplotlib.pyplot.xlabelвстановлює мітку осі x, тоді як matplotlib.pyplot.ylabelвстановлює мітку осі y поточної осі.

Код рішення:

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)

Вихідна цифра:

Ви також можете встановити заголовок вашої діаграми, додавши параметр заголовка наступним чином

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