построение графиков sub_plots из сводной таблицы с помощью matplotlib / seaborn

Question

У меня есть код из фрейма данных

Y = df['label']
for col in categorical_cols:
    tab = pd.crosstab(df[col],Y)
    annot = x.div(x.sum(axis=1).astype('float64'),axis=0)
    annot.plot(kind='bar',stacked=True)
    plt.title('Distribution of %s'%col)
    plt.xlabel('%s'%col,size='x-large')
    plt.xticks(rotation=45)
    plt.legend()

Как я могу построить их, используя разные субплоты в одном figure, потому что этот цикл печатает рисунок последнего столбца. Так что все цифры одинаковы.

Также: Как я могу произвести то же самое, используя matplotlib / seaborn, используя matplotlib, который показывает мне% или абсолютные значения.

Answer 1

Вам необходимо создать различные дочерние участки и затем передавать один объект axes на каждый вызов annot.plot с помощью ключевого слова ax, что-то вроде этого:

import math
import matplotlib.pyplot as plt

n = len(categorical_cols)
nrows = math.ceil(float(n) / 3.0)
fig, ax = plt.subplots(ncols=3, nrows=nrows, figsize=(9, nrows*3))
ax = ax.flatten()

Y = df['label']
for idx, col in enumerate(categorical_cols):
    tab = pd.crosstab(df[col],Y)
    annot = x.div(x.sum(axis=1).astype('float64'),axis=0)
    annot.plot(kind='bar',stacked=True, ax=ax[idx])
    ax[idx].title('Distribution of %s'%col)
    ax[idx].set_xlabel('%s'%col,size='x-large')
    ax.tick_params('x', labelrotation=45)
    plt.legend()