3D绘图线性回归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()

我希望它有所帮助