如何用Matplotlib绘制从数据点指向直线的箭头？

需要实现从每个散点指向直线y=x的垂直箭头，以表示点到直线的距离，但自行编写的代码箭头方向错误，且视觉上数据点出现偏移。

初始代码 #

import matplotlib.pyplot as plt
import numpy as np

# 创建45度直线
x_line = np.linspace(0, 10, 100)
y_line = x_line

# 生成直线附近的随机点
num_points = 20
x_points = np.linspace(2, 8, num_points)
y_points = x_points + np.random.normal(0, 0.5, num_points)

# 绘制直线和散点
plt.plot(x_line, y_line, label='Line', color='blue')
plt.scatter(x_points, y_points, label='Points', color='red')

# 设置图表标签和标题
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot Around a Line')

# 显示图例
plt.legend()

# 展示图表
plt.show()

尝试的错误代码 #

import matplotlib.pyplot as plt
import numpy as np

# 创建45度直线
x_line = np.linspace(0, 10, 100)
y_line = x_line

# 生成直线附近的随机点
num_points = 20
x_points = np.linspace(2, 8, num_points)
y_points = x_points + np.random.normal(0, 0.5, num_points)

# 绘制直线和散点
plt.plot(x_line, y_line, label='Line', color='blue')
plt.scatter(x_points, y_points, label='Points', color='red')

# 尝试添加从点到直线的箭头
for x, y in zip(x_points, y_points):
    plt.arrow(x, y, 0, y - x, color='black', linestyle='dashed', linewidth=0.5, head_width=0.2)

# 设置图表标签和标题
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot Around a Line')

# 显示图例
plt.legend()

# 展示图表
plt.show()

错误原因 #

1. 箭头方向逻辑错误：plt.arrow(x, y, dx, dy)的dx和dy是起点(x,y)的偏移量，而非终点坐标。原代码中dy = y - x，当y > x时箭头向上远离直线，y < x时向下远离，方向完全相反。
2. 未计算垂直垂足：原代码直接沿y轴偏移，并非指向点到直线的垂直投影点，无法表示真实的垂直距离。

正确实现代码 #

对于直线y = x，任意点(x, y)到该直线的垂直垂足坐标为((x+y)/2, (x+y)/2)。通过计算这个垂足，再设置正确的偏移量即可实现需求：

import matplotlib.pyplot as plt
import numpy as np

# 创建45度直线
x_line = np.linspace(0, 10, 100)
y_line = x_line

# 生成直线附近的随机点
num_points = 20
x_points = np.linspace(2, 8, num_points)
y_points = x_points + np.random.normal(0, 0.5, num_points)

# 绘制直线和散点
plt.plot(x_line, y_line, label='Line', color='blue')
plt.scatter(x_points, y_points, label='Points', color='red')

# 添加垂直箭头：从点指向直线的垂足
for x, y in zip(x_points, y_points):
    # 计算垂足坐标
    foot_x = (x + y) / 2
    foot_y = (x + y) / 2
    # 计算偏移量：垂足坐标减去点坐标
    dx = foot_x - x
    dy = foot_y - y
    # 绘制箭头
    plt.arrow(x, y, dx, dy, color='black', linestyle='dashed', linewidth=0.5, head_width=0.15)

# 设置图表属性
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.title('Scatter Plot with Perpendicular Arrows to Line')
plt.legend()

plt.show()

说明 #

• 垂足计算是核心：对于一般直线ax + by + c = 0，可以使用通用垂足公式计算，此处针对y=x（即x - y = 0）做了简化。
• head_width参数可根据图表尺寸调整，保证箭头显示清晰。

内容的提问来源于stack exchange，提问作者Alexander Obidiegwu