Как я могу пометить линии моего графика с помощью легенды, если они сделаны из .unstack ()? - PullRequest
0 голосов
/ 24 мая 2019

У меня есть развернутый кадр данных, который выглядит как

        accurate_prediction
year_week   actualscore 
2018_40     High    0.976554
            Low     0.403885
            Medium  0.132843
2018_41     High    0.978384
            Low     0.416383
            Medium  0.136292

и я хочу построить график year_week с 3 линиями при отображении их действительного значения с легендой оси

fig = plt.figure(num=None, figsize=(16, 10), facecolor='w', edgecolor='k')
ax = plt.axes()
plt.xticks(rotation=70)
ax.yaxis.set_major_locator(MaxNLocator(20))
ax.plot(prediction_over_time.unstack('actualscore'))
ax.grid()
ax.legend()

создает график, который я хочу, с 3 строками, с указанием недели года в сравнении с фактическим значением

Однако ax.legend () говорит мне, что у меня нет ярлыков. Обычно, когда я делаю этикетки, я говорю что-то вроде line=plot(df.column, label='example), но так как я делаю 3 строки из стека, у меня не может быть таких четких этикеток?

...