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

希望对你有帮助