问题描述
我正在尝试使用 Plotly 中的三个一维浮点数组创建一个 3D 曲面图。这是我的数据的样子:
alpha_1 alpha_2 sse
0 0.01 0.01 4.817321e+07
1 0.01 0.11 4.854821e+07
2 0.01 0.21 4.864922e+07
3 0.01 0.31 4.869423e+07
4 0.01 0.41 4.870884e+07
... ... ... ...
95 0.91 0.51 4.859500e+07
96 0.91 0.61 4.857550e+07
97 0.91 0.71 4.855622e+07
98 0.91 0.81 4.853898e+07
99 0.91 0.91 4.852403e+07
我想创建一个基本上模仿这个 3d 图的表面图。 这是我用来生成 3D 图的代码: see here 3D plot I want to make as surface plot
x = np.asarray(surface_df['alpha_1'])
y = np.asarray(surface_df['alpha_2'])
z = np.asarray(surface_df['sse'])
fig = plt.figure()
ax = plt.axes(projection="3d")
ax.scatter3D(x,y,z,c=z);
plt.show()
但这是我用 Plotly 得到的表面图。 Surface plot from Plotly.
这是我用于 plotly 的代码。
fig = go.figure(data=[go.Surface(x=surface_df['alpha_1'].values,y=surface_df['alpha_2'].values,z=surface_df[['alpha_1','sse','alpha_2']].values)])
fig.update_layout(title='plot',autosize=True,width=500,height=500)
fig.show()
如有任何指点,将不胜感激,谢谢!
解决方法
我认为您需要创建二维数组才能绘制曲面。
import pandas as pd
import numpy as np
import plotly.graph_objects as go
from scipy.interpolate import griddata
#Read data
data = pd.read_csv('your_data_set.txt',header=0,delimiter='\t')
#Create meshgrid for x,y
xi = np.linspace(min(data['alpha_1']),max(data['alpha_1']),num=100)
yi = np.linspace(min(data['alpha_2']),max(data['alpha_2']),num=100)
x_grid,y_grid = np.meshgrid(xi,yi)
#Grid data
z_grid = griddata((data['alpha_1'],data['alpha_2']),data['sse'],(x_grid,y_grid),method='cubic')
# Plotly 3D Surface
fig = go.Figure(go.Surface(x=x_grid,y=y_grid,z=z_grid,colorscale='viridis'))
fig.show()