目录

问题提出

绘制简单曲面图

使用自有数据拟合绘制

问题提出

在网上找了很久怎么用自己有的dataframe数据，拟合出3d的曲面图，大部分人都是根据已知函数去绘制，有几篇进行拟合绘制，但是还是根据特定函数生成的数据，没有一篇直接用dataframe数据画3d曲面图的。讲的比较好的几篇这里列出来供大家参考，本文主要针对使用自有数据拟合绘制3d曲面图进行详解。

python使用numpy和scipy模块拟合任意函数，多维曲面【转载】 – 牛哥是条狗 – 博客园 (cnblogs.com)https://www.cnblogs.com/niuniu238/articles/13551676.htmlscipy.optimize.curve_fit 拟合多维曲面问题_紫金山赵火龙的博客-CSDN博客https://blog.csdn.net/qq_26751117/article/details/106335185

绘制简单曲面图

为了由浅入深的讲解，这里首先介绍绘制一个简单的3维曲面的方法：

1、使用mpl_toolkits.mplot3d中的Axes3D进行三维绘图

2、使用numpy中的arange或linspace产生X轴坐标点和Y坐标点，并用meshgrid生成网格型数据，如果原本的X，Y按X=np.arange(0,4,1)，Y=np.arange(0,3,1)产生，那么网格化后X和Y如下：

# Xarray([[0., 1., 2., 3., 4.], [0., 1., 2., 3., 4.], [0., 1., 2., 3., 4.], [0., 1., 2., 3., 4.]])# Yarray([[0., 0., 0., 0., 0.], [1., 1., 1., 1., 1.], [2., 2., 2., 2., 2.], [3., 3., 3., 3., 3.]])

3、原本X和Y是一维，网格化后X和Y变成二维，根据二维的X和Y产生的Z数据也是二维的，最后使用plt的plot_surface绘制三维曲面，注意plot_surface传入的X,Y,Z数据必须是二维的

from matplotlib import pyplot as pltimport pandas as pd import numpy as npfrom scipy.interpolate import griddatafrom mpl_toolkits.mplot3d import Axes3D%matplotlib notebookfig=plt.figure()ax=Axes3D(fig)X=np.arange(-4,4,0.2)Y=np.arange(-4,4,0.2)X,Y=np.meshgrid(X,Y)Z=np.sqrt(X**2+Y**2)ax.plot_surface(X,Y,Z,rstride=1,cstride=1,cmap="rainbow")plt.draw()

使用自有数据拟合绘制

上面三维曲面图中的Z坐标数据是根据我们自己定义的函数计算的，但是在自有数据中，我们并不知道明确的函数，因此就需要用到scipy.optimize中的curve_fit拟合我们数据中的函数，再基于这个函数产生Z坐标数据进行绘制。绘制步骤如下：

1、根据你的先验知识或散点图情况，了解你的数据大致是什么函数情况，然后定义这个函数，我这里假设数据是一个二次曲面，即，那么需要定义一个这样的函数：

def func1(x,a,b,c,d,e,f):y=a*x[0]*x[0]+b*x[1]*x[1]+c*x[0]*x[1]+d*x[0]+e*x[1]+freturn y

2、将自有数据和定义的函数传入curve_fit中进行拟合，返回拟合参数和参数协方差，这里我使用自己数据中的a1和r1作为x[0]和x[1]，商家收益1作为y。

data=pd.read_csv(r'C:\Users\Desktop\汇总.csv',encoding='gbk')dt=datax1=np.array(dt['a1']).reshape(1,-1)#变成二维数据x2=np.array(dt['r1']).reshape(1,-1)#变成二维数据y=np.array(dt['商家收益1'])x=np.append(x1,x2,axis=0)#变成两行的二维数据popt, pcov =curve_fit(func1,x,y)

3、后续和前面绘制的步骤类似，只是Z坐标数据使用我们拟合的函数产生

X = np.linspace(0, 0.3, 15)Y = np.linspace(0, 1, 50)X, Y = np.meshgrid(X, Y)XX = np.expand_dims(X,axis=0)#变成三维数据，因为下面要把XX和YY合并到一起YY = np.expand_dims(Y,axis=0)x=np.append(XX,YY,axis=0)Z=func1(x,*popt)

4、最后使用plot_surface绘制三维曲面

完整的代码如下：

import pandas as pdimport numpy as npimport seaborn as snsimport matplotlib.pyplot as pltfrom scipy.interpolate import griddatafrom mpl_toolkits.mplot3d import Axes3Dfrom scipy.optimize import curve_fit# matplotlib 图形正常显示中文及负号plt.rcParams['font.sans-serif'] = ['SimHei'] # 用来正常显示中文标签plt.rcParams['axes.unicode_minus'] = False# 用来正常显示负号#3d绘图内嵌可旋转#%matplotlib notebook#3d绘图单独窗口#%matplotlibdef func1(x,a,b,c,d,e,f):z=a*x[0]*x[0]+b*x[1]*x[1]+c*x[0]*x[1]+d*x[0]+e*x[1]+freturn zdata=pd.read_csv(r'C:\Users\Desktop\汇总.csv',encoding='gbk')dt=datax1=np.array(dt['a1']).reshape(1,-1)#变成二维数据x2=np.array(dt['r1']).reshape(1,-1)#变成二维数据y=np.array(dt['商家收益1'])x=np.append(x1,x2,axis=0)#变成两行的二维数据popt, pcov =curve_fit(func1,x,y)X = np.linspace(0, 0.3, 15)Y = np.linspace(0, 1, 50)X, Y = np.meshgrid(X, Y)XX = np.expand_dims(X,axis=0)#变成三维数据，因为下面要把XX和YY合并到一起YY = np.expand_dims(Y,axis=0)x=np.append(XX,YY,axis=0)Z=func1(x,*popt)fig=plt.figure(figsize=(10,8))ax=plt.axes(projection="3d")ax3d=ax.plot_surface(X, Y, Z, rstride=1, cstride=1, cmap='jet')plt.show()

根据真实数据绘制的三维曲面图如下：