我们直接上代码了,之前已经介绍了如何导入表格了,现在就用表格中的数据来进行画图:还是用之前的表格
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()
结果图如下:
如果有哪里写的不对,就指出来哦。博主会虚心接受的。