在Plotly中创建三维曲面打印

Question

我想通过从外部文件读取数据，在Plotly中创建一个3D曲面图。下面是我使用的代码：

    import numpy as np
    import plotly.graph_objects as go 
    import plotly.express as px
    
    
    data = np.genfromtxt('values.dat', dtype=float)
    
    # The shape of X, Y and Z is (10,1)
    X = data[:,0:1] 
    Y = data[:,1:2]
    Z = data[:,2:3]
    
    fig = go.Surface(x=X, y=Y, z=Z, name='Surface plot', colorscale=px.colors.sequential.Plotly3)
    plot(fig)

上面的代码不会生成任何曲面图。必须进行哪些更改才能创建曲面打印？

Answer 1

从plotly reference：

描述曲面坐标的数据在z中设置。z中的数据应该是2D列表。x和y中的坐标可以是一维列表或{2D数组}

我有一个示例数据集。它包含三列(x，y，z)。

import plotly.graph_objects as go
import pandas as pd
import numpy as np
from scipy.interpolate import griddata


df = pd.read_csv('./test_data.csv')

x = np.array(df.lon)
y = np.array(df.lat)
z = np.array(df.value)


xi = np.linspace(x.min(), x.max(), 100)
yi = np.linspace(y.min(), y.max(), 100)

X,Y = np.meshgrid(xi,yi)

Z = griddata((x,y),z,(X,Y), method='cubic')



fig = go.Figure(go.Surface(x=xi,y=yi,z=Z))
fig.show()

参考页面：https://plotly.com/python/reference/surface/

​

​