Matplotlib：将折线图缩小为特定性别

Question

我有一个包含3列的DataFrame：'values'，'date'，'gender'。我有一些代码可以生成两列的折线图。

代码

f = plt.figure(figsize=(14,6))

plt.plot(twelve_week_program.groupby(['date'])['values'].mean(), linewidth=5)

我想通过绘制两条不同的线来包括'gender'列：一条用于男性，另一条用于女性。目前的代码未考虑性别。它只是绘制每行。

理想的输出

相同的情节，只有两个折线图-每个性别一个。

Answer 1

我将DataFrame的索引设置为“日期”列，然后使用：

plt.plot(twelve_week_program.values[twelve_week_program.gender=='F']), linewidth=5)
plt.plot(twelve_week_program.values[twelve_week_program.gender=='M']), linewidth=5)
plt.show()

提供“ F”和“ M”就是您表示该列的方式

Answer 2

我的答案：

plt.plot(df.groupby(
[df['date'], 'gender'])['values'].mean().unstack(), linewidth=5)