python - 使用 python 的 matplotlib 向散点图添加线

Question

我正在使用 python 的 matplotlib并想创建一个带有附加行的 matplotlib.scatter()。该线应从左下角到右上角，与散点内容无关。通过数据进行线性回归，如 this post , 不是我要找的。此外，它应该是动态的并且独立于散点输入。

这应该是最终情节:

编辑:

这样做得到了结果:

# Scatter Plot
x = data_calc_hourly.temp
y =  data_obs_hourly.temp

lineStart = data_calc_hourly.temp.min() 
lineEnd = data_calc_hourly.temp.max()  

plt.figure()
plt.scatter(x, y, color = 'k', alpha=0.5)
plt.plot([lineStart, lineEnd], [lineStart, lineEnd], 'k-', color = 'r')
plt.xlim(lineStart, lineEnd)
plt.ylim(lineStart, lineEnd)
plt.show()

有没有更好的办法？

Answer 1

这会绘制一条独立于散点图数据的对角线，即使您调整窗口大小时，它仍以轴为根:

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.lines as mlines
import matplotlib.transforms as mtransforms

x, y = np.random.random((2, 100))*2
fig, ax = plt.subplots()
ax.scatter(x, y, c='black')
line = mlines.Line2D([0, 1], [0, 1], color='red')
transform = ax.transAxes
line.set_transform(transform)
ax.add_line(line)
plt.show()