python绘制函数x^2+y^2的3D图像

短命女 2023-01-20 08:05 8阅读 0赞

### python绘制函数x^2+y^2的图像 ###

画出函数  f ( x , y ) = x 2 + y 2 f(x,y)=\{x\}^ \{2\}+\{y\}^ \{2\} f(x,y)=x2\+y2 的图像

meshgrid()函数：  
初始散点数据处理成xy网格数据

plot\_surface(x, y, f(x, y)，rstride=1, cstride=1, cmap=‘rainbow’)  
x、y数据决定坐标点，f(x,y)数据决定 x、y 坐标点对应的高度；rstride（row）指定行的跨度；cstride(column)指定列的跨度；cmap 设置颜色。

程序代码：

from matplotlib import pyplot as plt
    import numpy as np
    # 导入可以创建三维坐标的库
    from mpl_toolkits.mplot3d import Axes3D
    # 设置字体，使可以输入中文
    plt.rcParams['font.sans-serif']=['STSong']
    # 绘制三维坐标
    # 创建一个新的matplotlib.figure.Figure并为其添加一个类型为Axes3D的新轴
    fig = plt.figure()
    ax = Axes3D(fig)
    # 生成数据
    x = np.arange(-3, 3, 0.2)  # 从-3到3单位间隔为0.2
    y = np.arange(-3, 3, 0.2)  # 从-3到3单位间隔为0.2
    x, y = np.meshgrid(x, y)  # 初始散点数据处理成xy网格数据
    def fun(x, y):  # 定义函数
        f = x**2 + y**2  # 计算函数式
        return f
    f = fun(x, y)  # 函数调用
    # 绘制函数图像
    ax.set_xlabel('x')  # x轴
    ax.set_ylabel('y')  # y轴
    ax.set_zlabel('f')  # f轴
    # 标题名称、位置：0左下，1右上
    ax.text2D(0.3, 0.95, "函数f(x, y) = x^2 + y^2的函数曲线", transform=ax.transAxes)
    ax.plot_surface(x, y, f)
    plt.show()

画出函数的图像：  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80NTI3MjkwOA_size_16_color_FFFFFF_t_70]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L3dlaXhpbl80NTI3MjkwOA_size_16_color_FFFFFF_t_70]: /images/20221021/8dfcc6558bd34447b963b2629986f294.png