三维绘图线性回归pylab

Question

我需要绘制一个具有2个特征的线性回归问题。所以我认为在这种情况下，而不是一条线，我需要一个超平面来分隔我的数据。我已经完成了一个功能，但没有任何想法。

这是我的300个样本和1个特征的数据集图。

在此先感谢

Answer 1

检查numpy的meshgrid和plot_surface方法

尝试运行此示例代码，并将其转变为自己的喜好

from mpl_toolkits.mplot3d import Axes3D 
from matplotlib import cm 
import matplotlib.pyplot as plt 
import numpy as np 

fig=plt.figure() 
ax = Axes3D(fig) 

x=[1,2,3,4,5] 
y=[1,3,5,6,8] 
z=[3,3,5,6,7] 
x, y= np.meshgrid(x,y) 


ax.plot_surface(x,y,z) 

plt.show() 

我希望它能帮助