使用pandas和matplotlib绘制多个折线图

4 回答

• 14

  您可以使用类似下面的方法 . 您可以根据每个模板的值简单地切割数据框，然后使用图的日期和分数 .

  from pandas import *
  import matplotlib.pyplot as plt
  import matplotlib.dates as mdates
  import datetime as dt
  
  #The following part is just for generating something similar to your dataframe
  date1 = "20140605"
  date2 = "20140606"
  
  d = {'date': Series([date1]*5 + [date2]*5), 'template': Series(range(5)*2),
  'score': Series([random() for i in range(10)]) } 
  
  data = DataFrame(d)
  #end of dataset generation
  
  fig, ax = plt.subplots()
  
  for temp in range(5):
      dat = data[data['template']==temp]
      dates =  dat['date']
      dates_f = [dt.datetime.strptime(date,'%Y%m%d') for date in dates]
      ax.plot(dates_f, dat['score'], label = "Template: {0}".format(temp))
  
  plt.xlabel("Date")
  plt.ylabel("Score")
  ax.legend()
  plt.show()
  

  回复于 2024-05-08T03:26:20+08:00
• 2

  您可以使用groupby方法：

  data.groupby("template").plot(x="date", y="score")
  

  回复于 2024-05-08T03:26:20+08:00
• 52

  我认为使用同一图表上的所有行绘制此数据的最简单方法是将其旋转，使得每个“模板”值都是一列：

  pivoted = pandas.pivot_table(data, values='score', columns='template', index='date')
  # Now there will be an index column for date and value columns for 0,1,2,3,4
  pivoted.plot()
  

  回复于 2024-05-08T03:26:20+08:00
• 11

  您可以根据组添加图例：

  plt.legend(pr['template'], loc='best')
  

  回复于 2024-05-08T03:26:20+08:00