在python中使用matplotlib画函数图像

孙泽辉

编程

发布于：2021年7月3日

字数：139字

时长：1分钟

次浏览

最近学高数学上头了，如果把学到的高数知识在python中实现出来，那该有多酷！

生成$x^2$的函数图像python代码：

import numpy as np
import matplotlib.pyplot as plt


def main():
    # 生成 -5至-5 的x轴，步长为0.2
    x = np.array(np.arange(-5, 5.2, 0.2))
    # 自动映射，变成一个新数组返回
    y = x ** 2

    # 画坐标轴
    ax = plt.gca()  # 获取坐标轴
    ax.spines['right'].set_color('none')
    ax.spines['top'].set_color('none')
    ax.spines['bottom'].set_position(('data', 0))
    ax.spines['left'].set_position(('axes', 0.5))

    # 生成，展示
    plt.plot(x, y)
    plt.show()

成果：