Наложение строк в граф Matplotlib?

Question

Я пытаюсь напечатать простой линейный график, где вывод в отслеживании imdbRating во времени различных сезонов из шоу. Я ввел свой код так:

S1imdb = season_1['imdbRating']
S2imdb = season_2['imdbRating']
S3imdb = season_3['imdbRating']
S4imdb = season_4['imdbRating']
S5imdb = season_5['imdbRating']
S6imdb = season_6['imdbRating']
S7imdb = season_7['imdbRating']
S8imdb = season_8['imdbRating']
S9imdb = season_9['imdbRating']


plt.plot(S1imdb, label = "Season 1")
plt.plot(S2imdb, label = 'Season 2')
plt.plot(S3imdb, label = 'Season 3')
plt.plot(S4imdb, label = 'Season 4')
plt.plot(S5imdb, label = 'Season 5')
plt.plot(S6imdb, label = 'Season 6')
plt.plot(S7imdb, label = 'Season 7')
plt.plot(S8imdb, label = 'Season 8')
plt.plot(S9imdb, label = 'Season 9')
plt.show()

и мой вывод такой:

График рейтинга IMDB + Сезон Numebr

Однако я бы хотел, чтобы эти линии отображались друг на друга, а не рядом друг с другом. Как мне это сделать?

Answer 1

Вы можете создать новые координаты x из длин фреймов данных.

seasons = [season_1, season_2, season_3, season_4, ...]

for i, season in enumerate(seasons):
    y = season['imdbRating']
    x = range(1, len(y)+1)
    plt.plot(x, y, label = "Season {}".format(i+1))