Использовать имена столбцов dataframe в качестве меток в pylab.plot

Question

Я хотел бы построить данные в кадре данных и присвоить заголовкам столбцов метки. Я попытался это:

dfm.columns = ['a','b'] 
plot(dfm.cumsum(), label= dfm.columns.values) 
legend(loc='upper left') 

Но получилось так:

Вместо того, чтобы обе эти линии помечены [ «а», «б»], я хотел бы синюю линию, чтобы быть и зеленый цвет будет использоваться с помощью pylab

Answer 1

Я думаю, что у вас есть данные, созданные в части кода, который вы не показываете.

Вот пример, я использовал df.plot() в этом случае.

import pandas as pd 
import random 
import matplotlib.pyplot as plt 

x = [random.randint(10,20) for r in range(100)] 
y = [random.randint(0,10) for r in range(100)] 

df = pd.DataFrame([x,y]).T #T for transpose 

df.columns=['a','b'] 

df.plot(kind='line') 

plt.legend(loc='upper left') 

plt.show() 

Редактировать

pylab версия

import pandas as pd 
import random 
import matplotlib.pylab as plt 

x = [random.randint(10,20) for r in range(100)] 
y = [random.randint(0,10) for r in range(100)] 

df = pd.DataFrame([x,y]).T 

plt.plot(df) 

plt.legend(['a','b'],loc='upper left') 

plt.show()