Matplotlib绘图教程3

legend图例

要设置图例，得先给画出的两条线起“名字”，比如给第一条绘制的图线起名字为‘up’，给第二条绘制的图线起名字为‘down’。

plt.plot(x, y2, label='up')
plt.plot(x, y1,color='red',linewidth=1.0,linestyle='--',label='down')
plt.legend()

plt.legend()里的参数有handles、labels、loc（位置）
handle：plt.plot是有返回值的，例如给第一条线返回为l1，但是若想把这些对象传入到handle中，需要加逗号。如下所示：

l1,=plt.plot(x, y2, label='up')
l2,=plt.plot(x, y1,color='red',linewidth=1.0,linestyle='--',label='down')

labels：可以改变legend中对象的名字，不会用到原来设置的label。

l1,=plt.plot(x, y2, label='up')
l2,=plt.plot(x, y1,color='red',linewidth=1.0,linestyle='--',label='down')
plt.legend(handles=[l1,l2,],labels=['aaa','bbb'],loc='best')

如果只想显示一个图例，如只想显示aaa，就直接删掉’bbb’

annotation注释

原始图

先plot一个原始的图：

import matplotlib.pyplot as plt
import numpy as np

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

plt.figure(num = 1, figsize=(8, 5))
plt.plot(x, y)

ax = plt.gca()
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')

ax.xaxis.set_ticks_position('bottom')
ax.spines['bottom'].set_position(('data', 0))
ax.yaxis.set_ticks_position('left')
ax.spines['left'].set_position(('data', 0))

plt.show()

得到如下所示的图：

添加一种形式的标注（点)

plt.scatter()：这是一个返回点的函数。
如把上图的plt.plot(x, y)换成plt.scatter(x, y)，会得到如下所示的图：

• 用scatter展示一个点（x0，y0）：

x0 = 1
y0 = 2*x0+1
plt.scatter(x0, y0，s=50,color='b') #大小（size）是50，颜色是blue
plt.show()

• 画一条虚线

 plt.plot([x0, x0], [y0, 0],'k--',linewidth=2.5)

添加标注：

plt.annotate(r'$2x+1=%s$'%y0,xy=(x0,y0),xycoords='data',xytext=(+30,-30),textcoords='offset points',
        fontsize=16,arrowprops = dict(srrowstyle='->',connectionstyle='arc3, rad=.2'))

这里面的参数较多，作一些简要解释：
xycoords=‘data’ 表示这个标注是基于data，即基于这个点生成的。
xytext表示标注相对于这个点的位置。
得到的效果如下图所示：

添加第二种形式的标注（文本标注)

plt.text(-3.7,3,r'$This\ is\ the\ some\ text,\mu\ \sigma_i\ \alpha_t$',fontdict={'size':16,'color':'r'})

得到的标注图片效果如下：

附录（完整代码）

import matplotlib.pyplot as plt
import numpy as np

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

plt.figure(num = 1, figsize=(8, 5))
plt.plot(x, y)

ax = plt.gca()
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')

ax.xaxis.set_ticks_position('bottom')
ax.spines['bottom'].set_position(('data', 0))
ax.yaxis.set_ticks_position('left')
ax.spines['left'].set_position(('data', 0))

x0 = 1
y0 = 2*x0+1
plt.scatter(x0, y0)

plt.plot([x0, x0], [y0, 0],'k--',linewidth=2.5)

#method1
##########################################################
plt.annotate(r'$2x+1=%s$'%y0,xy=(x0,y0),xycoords='data',xytext=(+30,-30),textcoords='offset points',
        fontsize=16,arrowprops = dict(arrowstyle='->',connectionstyle='arc3, rad=.2'))

#method2
##########################################################

plt.text(-3.7,3,r'$This\ is\ the\ some\ text,\mu\ \sigma_i\ \alpha_t$',fontdict={'size':16,'color':'r'})

plt.show()