在线性坐标系中绘制一次函数图象

本文记述了用 Matplotlib 在线性坐标系中绘制一次函数图象的例子。

代码主体内容如下：

...

def main():

    fig, ax = plt.subplots(figsize=(8,8))                   #1

    ax = configure_axes(ax, 'Linear Function', 10, 10, 5, 1)          #2

    x = np.linspace(-4, 4, 100)             #7
    y = 2 * x + 3/2
    ax.plot(x, y, color='b')

    x = np.linspace(-9, 9, 100)
    y = - 1/2 * x - 3
    ax.plot(x, y, color='r')

    ax.text(5, 9,  r'$y = 2x + \frac{3}{2}}$',  color='b', horizontalalignment='left',  verticalalignment='bottom')
    ax.text(9, -8, r'$y = -\frac{1}{2}x - 3}$', color='r', horizontalalignment='right', verticalalignment='top')
                                                                #8
    fig.tight_layout()                  #9

    ...


def configure_axes(ax, title, xlimit, ylimit, , majorunit = 5, minorunit = 1):

    ax.set_title(title, fontsize='xx-large', loc='left')

    ax.spines['left'].set_position('zero')                      #3
    ax.spines['right'].set_visible(False)
    ax.spines['bottom'].set_position('zero')
    ax.spines['top'].set_visible(False)

    ax.text(-0.2, -0.2, 'O', fontsize='x-large', fontstyle='italic', horizontalalignment='right', verticalalignment='top')
                                                                            #4

    xlimit += 0.8                           #5
    ylimit += 0.8
    ax.set_xlim((-xlimit, xlimit))
    ax.set_ylim((-ylimit, ylimit))

    ax.plot(1, 0, '>k', transform=ax.get_yaxis_transform(), clip_on=False)              #6
    ax.xaxis.set_label_text('x')
    ax.xaxis.set_label_coords(1, 0.48)
    ax.xaxis.set_major_formatter(FuncFormatter(lambda w, pos: int(w) if w else ''))
    ax.xaxis.set_major_locator(MultipleLocator(majorunit))
    ax.xaxis.set_minor_locator(MultipleLocator(minorunit))
    ax.xaxis.set_tick_params(which='both', direction='in')

    ax.plot(0, 1, '^k', transform=ax.get_xaxis_transform(), clip_on=False)
    ax.yaxis.set_label_text('y')
    
    ...


if __name__ == '__main__': main()

在绘图前，准备一个特定大小的区域（#1）。配置坐标系（#2），其中主要包括：；将坐标系轴置为原点位于图象中心的样式（#3）；放置原点（#4）；控制两个坐标轴的刻度范围边界（#5）；设置 X 轴的坐标样式、名称、主次刻度值和位置（#6），Y 轴同样处理。绘图时，用蓝色实线和红色实线分别绘制 y = 2x + 3/2 和 y = -1/2x - 3 这两个一次函数图象（#7），并附上数学表达式说明（#8）。最后调整绘图区域大小以填充整个图象区域（#9）。

图象显示如下：

此代码可在 Matplotlib 3.3.4，Python 3.6.8 环境中运行。完整的代码请参考 [gitee] cnblogs/18435847 。

更多例子请参考 Matplotlib Gallery。