pandas 散点图 趋势线

在 Pandas 中绘制散点图,可以使用 Pandas 的 plot 方法。你可以通过设置 kind 参数为 'scatter' 来绘制散点图。具体代码如下:

import pandas as pd
import matplotlib.pyplot as plt
# 创建一个数据帧
df = pd.DataFrame({'x': [1, 2, 3, 4], 'y': [3, 4, 5, 6]})
# 绘制散点图
df.plot(kind='scatter', x='x', y='y')
plt.show()

如果你想绘制趋势线,可以使用 polyfit 函数计算出多项式拟合的系数,然后用 polyval 函数求出该拟合曲线。具体代码如下:

import numpy as np
# 计算多项式拟合的系数
z = np.polyfit(df['x'], df['y'], 1)
# 求出该拟合曲线
p = np.poly1d(z)
# 绘制散点图和趋势线
df.plot(kind='scatter', x='x', y='y')
plt.plot(df['x'], p(df['x']), 'r')
plt.show()
  •