python绘制3d动态模型_python – 在3D中使用Matplotlib绘制线性模型

我正在尝试创建适合数据集的线性模型的3d图.我能够在R中相对容易地做到这一点,但我真的很难在

Python中做同样的事情.这是我在R中所做的：

这是我在Python中所做的：

from mpl_toolkits.mplot3d import Axes3D

import matplotlib.pyplot as plt

import numpy as np

import pandas as pd

import statsmodels.formula.api as sm

csv = pd.read_csv('http://www-bcf.usc.edu/~gareth/ISL/Advertising.csv', index_col=0)

model = sm.ols(formula='Sales ~ TV + Radio', data = csv)

fit = model.fit()

fit.summary()

fig = plt.figure()

ax = fig.add_subplot(111, projection='3d')

ax.scatter(csv['TV'], csv['Radio'], csv['Sales'], c='r', marker='o')

xx, yy = np.meshgrid(csv['TV'], csv['Radio'])

# Not what I expected :(

# ax.plot_surface(xx, yy, fit.fittedvalues)

ax.set_xlabel('TV')

ax.set_ylabel('Radio')

ax.set_zlabel('Sales')

plt.show()

我做错了什么,我该怎么做？

谢谢.