使用matplotlib.pyplot在读入的图片上画线段

使用 matplotlib 在读入的图片上画线，关键点在于图片和线段共用一个坐标 ax，先使用 imshow 函数来显示图片，再使用 plot 函数来在图片上画线

import matplotlib.pyplot as plt  
import matplotlib.image as mpimg  
import numpy as np  
  
# 读取图片  
img = mpimg.imread('path_to_image.jpg')  # 替换为图片路径  
  
# 创建一个图形和轴  
fig, ax = plt.subplots()  
  
# 显示图片  
ax.imshow(img)  
   
  
# 定义线的起点和终点坐标（在图片坐标系中）  
# 这里假设要从图片的左上角画到右下角  
start_point = (0, 0)  
end_point = (img.shape[0], img.shape[1])  
  
# 在图片上画线  
ax.plot([start_point[0], end_point[0]], [start_point[1], end_point[1]], 'r-', linewidth=2)  # 红色实线，线宽为2  
  
# 关闭坐标轴（可选）  
ax.axis('off')  
  
# 显示图形  
plt.show()