李沐d2l包里重要的绘图函数理解

这个贴子主要记录李沐动手学深度学习里边的那个绘图函数。

绘图1

在预备知识章节的微积分章节，第一次提出了绘图函数：

def use_svg_display():  #@save
    """使用svg格式在Jupyter中显示绘图"""
    backend_inline.set_matplotlib_formats('svg')

def set_figsize(figsize=(3.5, 2.5)):  #@save
    """设置matplotlib的图表大小"""
    use_svg_display()
    d2l.plt.rcParams['figure.figsize'] = figsize

#@save
def set_axes(axes, xlabel, ylabel, xlim, ylim, xscale, yscale, legend):
    """设置matplotlib的轴"""
    axes.set_xlabel(xlabel)
    axes.set_ylabel(ylabel)
    axes.set_xscale(xscale)
    axes.set_yscale(yscale)
    axes.set_xlim(xlim)
    axes.set_ylim(ylim)
    if legend:
        axes.legend(legend)
    axes.grid()

#@save
def plot(X, Y=None, xlabel=None, ylabel=None, legend=None, xlim=None,
         ylim=None, xscale='linear', yscale='linear',
         fmts=('-', 'm--', 'g-.', 'r:'), figsize=(3.5, 2.5), axes=None):
    """绘制数据点"""
    if legend is None:
        legend = []

    set_figsize(figsize)
    axes = axes if axes else d2l.plt.gca()

    # 如果X有一个轴，输出True
    def has_one_axis(X):
        return (hasattr(X, "ndim") and X.ndim == 1 or isinstance(X, list)
                and not hasattr(X[0], "__len__"))

    if has_one_axis(X):
        X = [X]
    if Y is None:
        X, Y = [[]] * len(X), X
    elif has_one_axis(Y):
        Y = [Y]
    if len(X) != len(Y):
        X = X * len(Y)
    axes.cla()
    #通过数据在空白画布轴上画画，（x数据，y数据，线的样式）
    for x, y, fmt in zip(X, Y, fmts):
        if len(x):
            axes.plot(x, y, fmt)
        else:
            axes.plot(y, fmt)
    #将空白画布裱进画框里，设置画框的参数（画布，x轴名称，y轴名称，x轴的范围，y轴的范围，线的名称）
    set_axes(axes, xlabel, ylabel, xlim, ylim, xscale, yscale, legend)

主要包含以下部分：

• set_figsize()
  设置最终显示在屏幕上的图片的大小和格式
• set_axes()
  设置“相框”坐标轴
• plot
  主要的绘制主函数

主要介绍plot函数，绘图主要包含以下几步：

1. 设置显示在屏幕上图片的大小和格式
2. 获得绘画的主体——axes
3. axes上画画，要求有“自变量”、“因变量”、“线的样式”
4. 设置axes，设置x轴y轴坐标名称，x轴y轴刻度，x轴y轴的缩放，绘制线段的名称