Python基础教程之Matplotlib-图例

>>>接上一节内容，继续学习Matplotlib的基本绘图功能。

import matplotlib.pyplot as plt
import numpy as np

#定义函数


x = np.linspace(3,-3,30) 
y1 = 2*x + 1
y2 = x**2

#设置图像窗口 两项参数可缺省
plt.figure(num=2,figsize=(6,5))
#绘图 可设置曲线颜色、曲线宽度、曲线类型、曲线标签等
plt.plot(x,y1,color=orange,linestyle=-.，label=linear line)
plt.plot(x,y2,label=square line)
#设置横纵坐标轴的值范围
plt.xlim(-1,1)
plt.ylim(-2,4)
#设置横纵轴的名称
plt.xlabel()
plt.ylabel()
#获取当前坐标轴信息
ax = plt.gca() 
#设置坐标轴边框
ax.spines[right].set_color(none)
ax.spines[top].set_color(none)
ax.xaxis.set_ticks_position(bottom)
#spines设置边框
ax.spines[bottom].set_position((data,0))
ax.yaxis.set_ticks_position(left)
ax.spines[left].set_position((data,0))
#legend设置图例 其中参数loc设置位置，参数值可选upper left/right等
plt.legend()
plt.show()

>>>添加标记点plt.scatter() 注释 annotate

import matplotlib.pyplot as plt
import numpy as np

x=np.linspace(-3,3,50)
y = x + 1

plt.figure(num=2,figsize=(8,5)) #设置图表大小
plt.plot(x,y,linewidth=1,linestyle=-) #绘图并设置线宽等格式
plt.xlim(-1,2) #设置横坐标范围
plt.ylim(-1,2) #设置纵坐标范围
#plt.xticks([-1.0,-0.5,1],[r$bad$,r$normal$,r$good$])
ax = plt.gca() #坐标轴边框
ax.spines[top].set_color(none)
ax.spines[right].set_color(none)
ax.spines[bottom].set_position((data,0))
ax.spines[left].set_position((data,0))
x0=0.5
y0=x0+1
plt.plot([x0,x0],[y0,0],linestyle=-.)
plt.scatter(x0,y0)
plt.annotate(r$x0+1=%s$ %y0,xy=(x0,y0),xycoords=data,xytext=(+30,-30),
             textcoords=offset points, fontsize=16,arrowprops=dict(arrowstyle=->, connectionstyle="arc3,rad=.2"))
plt.show()