python-3.x - 从 Pandas 数据框创建多行时遇到问题

Question

我正在尝试创建具有多条线的线图。

我的数据框如下所示：

UNIVERSITY YEAR RECORDS
BROWN 1995-01-01 100
BROWN 2000-01-01 1000
YALE 1995-01-01 500
YALE 2000-01-01 5000

我一直在尝试的代码如下所示：

df = pd.read_csv('stats_test.txt',delimiter='\t',index_col='UNIVERSITY',parse_dates=['YEAR'])
lines = df.plot.line(x='YEAR',y='RECORDS',label=df['UNIVERSITY'])
plt.show()

我不知道如何为布朗和耶鲁找到一条线。任何帮助是极大的赞赏。谢谢你。

标签： python-3.xpandasmatplotlib

Answer 1

设置YEAR为索引，然后使用Groupby：

df.set_index('YEAR').groupby('UNIVERSITY')['RECORDS'].plot()

输出：