matplotlib legend图例&matplotlib标注

import matplotlib.pyplot as plt
import numpy as np

new_ticks = np.linspace(-2,2,11)
print(new_ticks)

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

#xy范围
plt.xlim((-1,2))
plt.ylim((-2,3))

#xy描述
plt.xlabel('xxxx')
plt.ylabel('yyyy')

plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')
#color指的是线的颜色，linewidth指的是线的粗细程度
#linestyle指的是线的类型
#第一个两个-指的就是虚线，第一个一个-指的就是实线

plt.xticks(new_ticks)
plt.yticks([-1,0,1,2,3],
          ['level1','level2','level3','level4','level5'])
plt.show()

安放图例

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

#xy范围
plt.xlim((-1,2))
plt.ylim((-2,3))

#xy描述
plt.xlabel('xxxx')
plt.ylabel('yyyy')

l1, = plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')
l2, = plt.plot(x,y2,color='blue',linewidth=5.0,linestyle='-')
#color指的是线的颜色，linewidth指的是线的粗细程度
#linestyle指的是线的类型
#第一个两个-指的就是虚线，第一个一个-指的就是实线
plt.legend(handles=[l1,l2],labels=['test1','test2'],loc='best')
#loc='best'这样函数会自动帮你找到最好的位置来放图例

plt.xticks(new_ticks)
plt.yticks([-1,0,1,2,3],
          ['level1','level2','level3','level4','level5'])
plt.show()

如何给图加上标注呢？

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(-1,1,100)
y1 = 2*x + 1

plt.plot(x,y1,color='red',linewidth=1.0,linestyle='--')

#gca代表获取当前坐标轴
ax = plt.gca()
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
#右边和上边的边框去掉

#把x轴的刻度设置为'bottom'，x轴刻度上的字都在x轴下部
#把y轴的刻度设置成'left'，y轴刻度上的字都在y轴左部
ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')

#设置x,y轴位置，让x轴y轴位于0的位置
ax.spines['bottom'].set_position(('data',0))
ax.spines['left'].set_position(('data',0))

x0 = 0.5
y0 = 2*x0 +1
#画点
plt.scatter(x0,y0,s=50,color='b')
#画虚线
#从（x0,y0）到（y0,0）画一条黑色的（k）虚线(--)
#线的宽度lw为2
plt.plot([x0,x0],[y0,0],'k--',lw=2)

#加上描述
#%后面描述的是位置，从x0,y0出发，x0+30,y0-30的位置
# r'$   $'中间加上描述的内容
#fontsize=16字体大小设为16
#arrowprops表示加上箭头，后面跟的是箭头格式描述，指箭头弧度是0.2
plt.annotate(r'$2x+1=%s$' % y0,xy=(x0,y0),xytext=(+30,-30),
             textcoords='offset points',fontsize=16,
            arrowprops=dict(arrowstyle='->',
             connectionstyle='arc3,rad=.2'))

#红色文字描述
plt.text(-1,2,r'$this\ is\ the\ text$',fontdict={'size':'16',
                                                 'color':'r'})

plt.show()