Matplotlib бок о бок

Question

Я пытаюсь построить следующий фрейм данных с помощью matplotlib:

df = pd.DataFrame({'X': ["A", "A", "B", "B"], 'Z': ["a", "b", "a", "b"], 'Y': [5, 1, 10, 5]})
df

    X   Z   Y
0   A   a   5
1   A   b   1
2   B   a   10
3   B   b   5

Мне нужны два столбца, где столбцы расположены рядом друг с другом, а не друг над другом. Когда я запускаю это, столбцы располагаются друг над другом:

plt.barh(df['X'][df['Z'] == "a"], df['Y'][df['Z'] == "a"], color = 'blue')
plt.barh(df['X'][df['Z'] == "b"], df['Y'][df['Z'] == "b"], color = 'red')

И когда я пытаюсь изменить положение столбцов, я получаю ошибку: can only concatenate str (not "float") to str. Как я могу обойти это?

Answer 1

Если вы хотите, чтобы планки располагались рядом, вы можете использовать библиотеку seaborn :

import seaborn as sns
sns.barplot(data=df, x='Y', hue='Z', y='X')

Answer 2

Использование DataFrame.pivot с DataFrame.plot.barh:

df.pivot(*df).plot.barh()

или

df.pivot(*df).plot(kind = 'bar')

Answer 3

Не совсем точно, что вы хотите, но вы можете попробовать это:

df.set_index(['X','Z'])['Y'].unstack().plot.barh()

Или

df.set_index(['X','Z'])['Y'].unstack().plot.bar()

Или

df.set_index(['X','Z'])['Y'].unstack().plot.barh(subplots=True, layout=(1,2))

или

df.set_index(['X','Z'])['Y'].unstack().plot.bar(subplots=True, layout=(1,2))