pytorch-matploblib 常用画图类型

import matplotlib.pyplot as plt
import numpy as np
# from numpy import power

"""画曲线图"""
# 生成画布
plt.figure(num=1, figsize=(8,5))
# numpy里有一个生成等差数列的函数
x = np.linspace(-10, 10, 200)
y1 = 2*x+3
y2 = 1.5*x**2+1
plt.plot(x,y1,color='r',linewidth=5, linestyle='-', marker='o', markersize='10', label='y1 = 2*x+3')
plt.plot(x,y2,color='b',linewidth=5, linestyle='--',label='y2 = 1.5*x**2+1')
x_ticks = list(np.linspace(-5,5,11))
y_ticks = list(np.linspace(-5,20,5))
# 更改x,y坐标轴的尺度标量区间
plt.xlim(-5,5)
plt.ylim(-5,20)
# 更改x,y坐标轴的尺度标记稀疏程度
plt.xticks(x_ticks)
plt.yticks(y_ticks)
# 创建图例的函数plt.legend()
plt.legend(loc='best')   # 自动寻找最适合的位置
# plt.legend(loc='upper center')  # 上中部
# plt.legend(loc='upper left')    # 左上部
# plt.legend(loc='lower left')    # 左下部
# plt.legend(loc='upper right')   # 右上部
# plt.legend(loc='lower right')   # 右下部
"""去掉图上边框，与右边框"""
ax = plt.gca()
ax.spines['right'].set_color('none')
ax.spines['top'].set_color('none')
plt.show()