如何在matplotlib.pyplot中格式化轴并包含图例?

时间:2013-07-25 12:14:09

标签: python matplotlib

获得图形的这个功能,想要格式化轴,使图形从(0,0)开始,我也如何写图例,这样我就可以标记哪条线属于y1,哪条线属于y2和标签轴。

  import matplotlib.pyplot as plt
  def graph_cust(cust_type): 
  """function produces a graph of day agaist customer number for a given customer type""" 
  s = show_all_states_list(cust_type)
  x = list(i['day']for i in s)
  y1 = list(i['custtypeA_nondp'] for i in s) 
  y2 = list(i['custtypeA_dp']for i in s) 
  plt.scatter(x,y1,color= 'k') 
  plt.scatter(x,y2,color='g') 
  plt.show() 

谢谢

1 个答案:

答案 0 :(得分:0)

您可以使用plt.xlim(x_low, x_high)在任一轴上设置限制。如果您不想手动设置上限(例如您对当前上限感到满意),请尝试:

ax = plt.subplot(111) # Create axis instance
ax.scatter(x, y1, color='k') # Same as you have above but use ax instead of plt
ax.set_xlim(0.0, ax.get_xlim()[1])

注意这里的细微差别,我们使用轴实例。这使我们能够使用ax.get_xlim()返回当前的xlimits,这会返回一个元组(x_low, x_high),我们使用[1]选择第二个。{/ p>

传奇的最小例子:

plt.plot(x,y,label =“some text”)    plt.legend()

有关传说的更多信息,请参阅any of these examples