Столбчатая диаграмма Python с группами на одном графике - PullRequest
Новая
       34

Столбчатая диаграмма Python с группами на одном графике

1 голос
/ 30 мая 2019

У меня есть такие данные: 

Group   yq        Value1    Value2
G       2014Q1     0.07        1.1
G       2014Q2     0.06        1.09
G       2014Q3     0.09        1.11
G       2014Q4     0.04        1.13
I       2014Q1     0.10        1.2
I       2014Q2     0.13        1.25
I       2014Q3     0.15        1.23
I       2014Q4     0.18        1.4

Я хочу построить линию и гистограмму на одном графике.
Сначала я попытался построить график, но он вывел два графика (2 группы, G и I): 

import matplotlib.pyplot as plt
ax = dataset.groupby('Group')[['yq', 'Value1']].plot(x = 'yq', kind='bar')

После этого я попытался нарисовать на нем линейный график. 

fig, ax1 = plt.subplots(figsize=(7, 5))
ax2 = ax1.twinx()
dataset[['Value1', 'yq', 'Group']].groupby('Group').plot(x = 'yq', kind='bar', color='y', ax=ax1)
dataset[['Value2', 'yq', 'Group']].groupby('Group').plot(x = 'yq', kind='line', marker='d', ax=ax2)
ax1.yaxis.tick_right()
ax2.yaxis.tick_left()

Однако сюжет странный. Не правильно отображать все метки на оси X. Быть кратким. это просто показывает год, а не год и квартал.
Более того, на графике также нет гистограммы.

Есть предложения?

Я тоже пробовал: 

fig, ax = plt.subplots(figsize=(10, 5))
dataset[['Value1', 'yq', 'Group']].plot(x = 'yq', kind='bar', stacked=False, title='get_title', color='grey', ax=ax, grid=False)
ax2 = ax.twinx()
ax2.plot(ax.get_xticks(), dataset[['Value2']].values, linestyle='-', marker='o', color='k', linewidth=1.0, label='percentage')
lines, labels = ax.get_legend_handles_labels()
lines2, labels2 = ax2.get_legend_handles_labels()
ax.legend(lines + lines2, labels + labels2, loc='best')
ax.yaxis.set_ticks_position("right")
ax2.yaxis.set_ticks_position("left")
fig.autofmt_xdate()
plt.show()

Этот график неправильный, но может отображать линии и столбцы на одном графике.

1 Ответ

2 голосов
/ 30 мая 2019

Проблема в том, что bar plot задает для оси x значение range(len(dataset)) и использует соответствующие метки, тогда как line plot этого не делает.Таким образом, вы можете изменить yq на строку и использовать seaborn: 

dataset.yq = dataset.yq.astype(str)

fig, ax1 = plt.subplots(figsize=(7,5))
ax2=ax1.twinx()
sns.barplot(x='yq', y='Value1', data=dataset, hue='Group',ax=ax1)
sns.lineplot(x='yq',y='Value2', data=dataset, hue='Group', marker='d', ax=ax2)
plt.show()

дает:

enter image description here

...