Как мне вставить имена диаграмм в качестве переменных из сгруппированного фрейма данных, используя python?

Question

Я новичок в Python и программировании в целом. Я пытаюсь создать несколько горизонтальных гистограмм из сгруппированного фрейма данных. Частично я добился успеха, используя следующий скрипт:

#building dataframes for data plots

plotdf = pd.DataFrame(merge_df, columns= ['ppset_id', 'Cell Line', 'Undil_CT'])

plotdf = plotdf.dropna()

plotdf = plotdf.reset_index(drop=True)


plotdf.groupby('ppset_id').plot(x="Cell Line", y="Undil_CT", kind = "barh", xlim=(0,45), figsize=(11,8), legend=None)

Результат почти полностью соответствует моему желанию с данными для каждого ppset_id в отдельной диаграмме, за исключением того, что в каждой диаграмме отсутствует заголовок диаграммы. Я пытаюсь установить заголовок диаграммы для каждого графика на уникальное значение ppset_id, связанное с нанесенными на график данными, но не смог найти способ сделать это после обширного поиска по форуму.

Заранее спасибо за вашу помощь.

Answer 1

Использование набора данных seaborn tips в качестве примера.

1. Сохранение всех осей в переменной (чтобы впоследствии вы могли получить доступ к каждой оси).
2. Группа доступа имена из DataFrame.groupby (column_name) .groups.keys ()
3. Оба имеют одинаковую длину, поэтому выполните итерацию по обоим и установите заголовок каждой оси для соответствующего имени группы

Пример кода с использованием набора данных seaborn tips:

axes = tips.groupby('sex').plot()
group_names = tips.groupby('sex').groups.keys()
for ax, title in zip(axes, group_names):
    ax.set_title(title)

Вывод:

введите описание изображения здесь