plotly 3维曲面

墨蓝 2023-06-25 02:29 25阅读 0赞

import numpy as np
    import plotly.graph_objs as go
    import matplotlib.pyplot as plt
    
    
    def f(x, y):
        z = np.power(x, 2) + np.power(y, 2)
        return z
    
    
    def fig():
        a = np.linspace(start=0, stop=3, num=4, dtype=np.int32)
        b = np.linspace(start=0, stop=3, num=4, dtype=np.int32)
        print(a)
        print(b)
        x, y = np.meshgrid(a, b)
        print(x)
        print(y)
        mesh(x, y)
    
        z = f(x, y)
        print(z)
    
        fig = go.Figure(data=[go.Surface(x=x, y=y, z=z)])
    
        fig.update_layout(
            title_text='3D',
            height=800,
            width=800,
            autosize=False,
            margin=dict(l=65, r=50, b=65, t=90)
        )
    
        fig.show()
    
    
    # 观察通过meshgrid形成的x-y坐标轴数据点。
    def mesh(x, y):
        plt.xlabel('x axis')
        plt.ylabel('y axis')
        plt.plot(x, y, marker='^', color='red', linestyle='none')
        plt.grid()
        plt.show()
    
    
    if __name__ == '__main__':
        fig()

生成的数据点：

[0 1 2 3]
    [0 1 2 3]
    [[0 1 2 3]
     [0 1 2 3]
     [0 1 2 3]
     [0 1 2 3]]
    [[0 0 0 0]
     [1 1 1 1]
     [2 2 2 2]
     [3 3 3 3]]
    [[ 0  1  4  9]
     [ 1  2  5 10]
     [ 4  5  8 13]
     [ 9 10 13 18]]

x-y坐标轴散点：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BoaWwuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70][]

最终的3D（3维）曲面：

![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BoaWwuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70 1][]

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BoaWwuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70]: https://img-blog.csdnimg.cn/2019122418313796.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly96aGFuZ3BoaWwuYmxvZy5jc2RuLm5ldA==,size_16,color_FFFFFF,t_70
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly96aGFuZ3BoaWwuYmxvZy5jc2RuLm5ldA_size_16_color_FFFFFF_t_70 1]: https://img-blog.csdnimg.cn/20191224183153876.png?x-oss-process=image/watermark,type_ZmFuZ3poZW5naGVpdGk,shadow_10,text_aHR0cHM6Ly96aGFuZ3BoaWwuYmxvZy5jc2RuLm5ldA==,size_16,color_FFFFFF,t_70