Проблемы с созданием нескольких строк из Dandas Dataframe

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()

Я не знаю, как получить строку для Брауна и линию для Йельского университета. Любая помощь очень ценится. Спасибо.

Answer 1

Установите YEAR в качестве индекса, а затем используйте Groupby :

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

Выход: