按数据框分组的两个不同的线图

Question

我已将数据分组到 2 个单独的数据框中，并希望将它们与 2 条单独的线一起绘制在一个图中。

我根据需要对数据进行了分组，并根据分组数据绘制了单独的图表。

grouped_men = df_men.groupby('age').mean()[['oldpeak']]
grouped_women = df_women.groupby('age').mean()[['oldpeak']]
grouped_men.plot(kind='line',title='Mens age vs oldpeak')
grouped_women.plot(kind='line',title='Womens age vs oldpeak')

但现在我需要绘制一个包含 2 条男性和女性线条的单一图表，而不是 2 个单独的图。

当前情节如下所示：

Answer 1

您需要指定 Pandas 放置绘图的坐标轴。请尝试以下操作：

import matplotlib.pyplot as plt
fig = plt.figure()
ax = fig.add_subplot(111)
grouped_men.plot(kind='line', ax=ax, label='Mens age vs oldpeak')
grouped_women.plot(kind='line', ax=ax, label='Womens age vs oldpeak')
plt.gca().legend(title="Legend title")  # Changes
plt.show()