matplotlib基本操作

import matplotlib.pyplot as plt
import numpy as np

写出两条线的方程，x是-3到3之间50个数的等差数列

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

定义一个figure下的所有操作都属于这个figure

plt.figure()
plt.plot(x, y1)

定义第二个figure

plt.figure(num=3, figsize=(8, 5))
plt.plot(x, y2)
plt.plot(x, y1, color='red', linewidth=1.0, linestyle='--')
plt.show()

对坐标轴的操作，还是上面的两条线

设置坐标轴的范围

plt.xlim((-1, 2))
plt.ylim((-2, 3))

增加坐标轴的描述

plt.xlabel('I am X')
plt.ylabel('I am Y')

对坐标轴的刻度进行替换

new_ticks = np.linspace(-1, 2, 5)
plt.xticks(new_ticks)
plt.yticks([-2, -1.8, 1, 3], ['really bad', 'bad', 'normal', 'nice'])

gca:get current axis
隐藏上面和右边的边界

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

将下边界和左边界设置为坐标轴

ax.xaxis.set_ticks_position('bottom')
ax.yaxis.set_ticks_position('left')

移动边界

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

Legend图例
在plt.plot中加上label标签
而legend函数中也可以加很多参数，如loc等

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

scatter代表画点，s代表size点的大小
下面用了简写，‘k–’代表黑色虚线，lw代表线的宽度

x0 = 1
y0 = 2 * x0 + 1
plt.scatter(x0, y0, s=50, color='b')
plt.plot([x0, x0], [y0, 0], 'k--',lw =2.5)

增加annotation
2x+1=3-----------------内容
xy=(x0, y0), xycoords=‘data’------------------以x0，y0为起点
xytext=(+30, -30),textcoords=‘offset points’-------------------text平移
arrowprops=dict(arrowstyle=’->’, connectionstyle=‘arc3,rad=.2’)---------设置那条弧线

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

另一种标注
仔细看text的格式，如何写一些特殊符号，空格

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