如何根据多列数据在 Pandas Python 中的一个图中绘制多条线？

Question

我有一个包含 3 列的数据框，如下所示：

import pandas as pd

y = [2005, 2005, 2005, 2015, 2015, 2015, 2030, 2030, 2030]
n = ['A', 'B', 'C', 'A', 'B', 'C', 'A', 'B', 'C']
w = [80, 65, 88, 65, 60, 70, 60, 55, 65]

df = pd.DataFrame({'year': y, 'name': n, 'weight': w})

   year name  weight
0  2005    A      80
1  2005    B      65
2  2005    C      88
3  2015    A      65
4  2015    B      60
5  2015    C      70
6  2030    A      60
7  2030    B      55
8  2030    C      65

我如何为 A、B 和 C 绘制一条线，显示它们的体重多年来的变化情况。所以我试过这个：

df.groupby("name").plot(x="year", y="weight")

但是，我得到了多个图，这不是我想要的。我希望所有这些情节都集中在一个数字中。

Answer 1

For this specific example a possible solution would be the following

df.set_index("year", inplace=True)
df.groupby("name")["weight"].plot(legend=True, xlabel="Year", ylabel="Weight")

Answer 2

这是否会产生您正在寻找的东西？

import matplotlib.pyplot as plt
fig,ax = plt.subplots()

for name in ['A','B','C']:
    ax.plot(df[df.name==name].year,df[df.name==name].weight,label=name)

ax.set_xlabel("year")
ax.set_ylabel("weight")
ax.legend(loc='best')