Добавьте метку и легенду для l oop для matplotlib

Question

введите описание изображения здесь

plt.figure(figsize = (8,5))
plt.title('Gas Prices over time in USD', fontdict={'fontname': 'Calibri', 'fontsize': 20})

for country in gas:
    if country != 'Year':
        plt.plot(gas.Year, gas[country], marker = '.')


plt.xticks(gas.Year[::3].tolist()+[2011])
plt.xlabel('Year')

plt.ylabel('US Dollars Per Gallon')

plt.show()

Я попытался добавить легенду на основе столбцов на моем matplotlib graph, я знаю, что мне нужно добавить метки, чтобы легенда отображалась. Это мой код, как мне это сделать в l oop. В частности эта часть:

for country in gas:
    if country != 'Year':
        plt.plot(gas.Year, gas[country], marker = '.', label = ?)

Answer 1

вы можете передать имя столбца в качестве метки в plt.plot. Затем вы можете вызвать plt.legend (), чтобы нарисовать легенду. Пример с фиктивными данными:

gas = pd.DataFrame({'Year': np.arange(10)+2010, 'Cameroon':np.random.randint(0,10,10), 'Nepal':np.random.randint(0,10,10)})


plt.figure(figsize = (8,5))
plt.title('Gas Prices over time in USD', fontdict={'fontname': 'Calibri', 'fontsize': 20})

for country in gas:
    if country != 'Year':
        plt.plot(gas.Year, gas[country], marker = '.', label=country)


plt.xticks(gas.Year[::3].tolist()+[2011])
plt.xlabel('Year')

plt.ylabel('US Dollars Per Gallon')
plt.legend()
plt.show()