Python使用Matpoltlib绘制该散点图一次和二次拟合线

42 阅读1分钟

引入依赖读取数据并绘制散点图

import matplotlib.pyplot as plt
import pandas as pd

plt.rcParams['font.size'] = 18
ax = plt.axes([0.1, 0.1, 0.8, 0.8])

# 读取数据,预处理
data = pd.read_csv('./study_performance.csv')
# 画散点图
plt.scatter(data['writing_score'], data['reading_score'], c='g', s=data["math_score"] * 10, alpha=0.2)
plt.title('writing_score,reading_score and math_score')
plt.xlabel('writing_score')
plt.ylabel('reading_score')

# 隐藏右上边框
ax.spines['top'].set_visible(False)
ax.spines['right'].set_visible(False)

plt.show()

效果图

散点图.png