python - 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()

我希望它有帮助