用python来做regression 图

我们直接上代码了，之前已经介绍了如何导入表格了，现在就用表格中的数据来进行画图：还是用之前的表格

import pandas as pd
import matplotlib.pyplot as plt

data = pd.read_csv('1.01. Simple linear regression.csv')

y = data['GPA']
x = data['SAT']

plt.scatter(x,y)
plt.xlabel('SAT', fontsize = 20)
plt.ylabel('GPA', fontsize = 20)
plt.show()

结果是这样的：

如果想在上面的图中画出回归方程的线时，就得将上面的代码改一下：

import pandas as pd
import matplotlib.pyplot as plt

data = pd.read_csv('1.01. Simple linear regression.csv')

y = data['GPA']
x1 = data['SAT']

plt.scatter(x1,y)
yhat = 0.0017*x1 + 0.275
fig = plt.plot(x1, yhat, lw=4, c='orange', label='regression line')
plt.xlabel('SAT', fontsize = 20)
plt.ylabel('GPA', fontsize = 20)
plt.show()

结果图如下：

如果有哪里写的不对，就指出来哦。博主会虚心接受的。