Pthon Matplotlib 画图

一、普通绘图

 import matplotlib.pyplot as plt
 import numpy as np

 # 绘制普通图像
 x = np.linspace(-1, 1, 50)
 y1 = 2 * x + 1
 y2 = x**2

 plt.figure()
 # 在绘制时设置lable, 逗号是必须的
 l1, = plt.plot(x, y1, label = 'line')
 l2, = plt.plot(x, y2, label = 'parabola', color = 'red', linewidth = 1.0, linestyle = '--')

 # 设置坐标轴的取值范围
 plt.xlim((-1, 1))
 plt.ylim((0, 2))

 # 设置坐标轴的lable
 plt.xlabel('X axis')
 plt.ylabel('Y axis')

 # 设置x坐标轴刻度, 原来为0.25, 修改后为0.5
 plt.xticks(np.linspace(-1, 1, 5))
 # 设置y坐标轴刻度及标签, $$是设置字体
 plt.yticks([0, 0.5], ['$minimum$', 'normal'])

 # 设置legend
 plt.legend(handles = [l1, l2,], labels = ['a', 'b'], loc = 'best')
 plt.show()

二、自定义单峰函数

 import math
 import numpy as np
 import matplotlib.pyplot as plt

 x = np.linspace(-30, 30, 500)
 y = []
 y2 = []
 a = 3
 b = 0
 c = 25
 for i in x :
     # 类似高斯函数，a 代表峰值， b对称轴位置，c方差
     temp = a * math.exp(-(i-b)**2 / (2 * c))
     y.append(temp)
     #对上一个单峰函数值进行放大处理,红色虚线部分
     y2.append(math.exp(temp))

 plt.figure()
 l1= plt.plot(x, y, label = 'line')
 l2, = plt.plot(x, y2, label = 'parabola', color = 'red', linewidth = 1.0, linestyle = '--')
 plt.show()

三、画subplot子图（2 x 2 为例）

     import matplotlib.pyplot as plt
     t=np.arange(0.0,2.0,0.1)
     s=np.sin(t*np.pi)
     plt.subplot(2,2,1) #要生成两行两列，这是第一个图plt.subplot('行','列','编号')
     plt.plot(t,s,'b--')
     plt.ylabel('y1')
     plt.subplot(2,2,2) #两行两列,这是第二个图
     plt.plot(2*t,s,'r--')
     plt.ylabel('y2')
     plt.subplot(2,2,3)#两行两列,这是第三个图
     plt.plot(3*t,s,'m--')
     plt.subplot(2,2,4)#两行两列,这是第四个图
     plt.plot(4*t,s,'k--')
     plt.show()

点图和线图

fig = plt.figure()
ax = fig.add_subplot(221, projection='3d')
ax.plot(array_normal[:,0],array_normal[:,1],array_normal[:,2])
plt.subplot(2,2,2)
plt.plot(np.arange(0,sample_len,1), signal_normal)

normal_pow = array_normal[:,2]

ax3 = fig.add_subplot(223, projection='3d')
ax3.plot(array_anomaly[:,0],array_anomaly[:,1],array_anomaly[:,2])

anomaly_pow = array_anomaly[:,2]
plt.subplot(2,2,4)
plt.scatter(np.arange(0,sample_len,1), signal_anomaly)
plt.show()

【Reference】

[1] https://www.jianshu.com/p/de223a79217a

[2] https://www.cnblogs.com/xingshansi/p/6777945.html