最小二乘法（csv切片实现可视化）

import matplotlib.pyplot as plt
import pandas as pd

rows = pd.read_csv('E:\chicago\d1.csv') 填写自己的csv地址
#print(rows)
#对数据切片


row_X = rows.iloc[0:11, 0:1]  # 0-11行  1-2列
print(row_X)
row_Y = rows.iloc[0:11, 1:2]
print(row_Y)

w= 0.5
# b=y（平均）-b*x（平均）
b= 7.5-w*9
#拟合函数
pred_y= w*row_X+b
plt.plot(row_X,pred_y,c='r')
plt.scatter(row_X,row_Y)
plt.savefig("E:\chicago\Figure\Figure1.jpg" )
plt.show()

实现图片

---