在 Matplotlib 中绘制两条线之间角度的最佳方法

matplotlibdata visualizationpython

在 Matplotlib 中绘制两条线之间角度的最佳方法是使用 Arc 类来制作一个角度弧来绘制它们之间的角度。

步骤

  • 设置图形大小并调整子图之间和周围的填充。
  • 使用 figure() 方法创建新图形或激活现有图形。
  • 使用 add_subplot() 方法将 '~.axes.Axes' 作为子图排列的一部分添加到图形中。
  • 创建 2D 线实例作为 l1 l2
  • 将线添加到当前轴。
  • 要绘制角度,请调用返回椭圆弧的用户定义方法。可以使用线的斜率创建弧长。
  • 使用 add_patch() 方法添加艺术家,即 arc
  • 要显示图形,请使用 show() 方法。

示例

from matplotlib import pyplot as plt, patches
import math

plt.rcParams["figure.figsize"] = [7.50, 3.50]
plt.rcParams["figure.autolayout"] = True

def angle_plot(line1, line2, offset=1.0, color=None, origin=(0, 0),
len_x_axis=1, len_y_axis=1):
   xy1 = line1.get_xydata()
   xy2 = line2.get_xydata()
   slope1 = (xy1[1][1] - xy1[0][1]) / float(xy1[1][0] - xy1[0][0])
   angle1 = abs(math.degrees(math.atan(slope1)))
   slope2 = (xy2[1][1] - xy2[0][1]) / float(xy2[1][0] - xy2[0][0])
   angle2 = abs(math.degrees(math.atan(slope2)))
   theta1 = min(angle1, angle2)
   theta2 = max(angle1, angle2)
   angle = theta2 - theta1
   if color is None:
      color = line1.get_color()

   return patches.Arc(origin, len_x_axis * offset, len_y_axis * offset, 0, theta1, theta2, color=color, label=str(angle) + u"\u00b0")

fig = plt.figure()
ax = fig.add_subplot(1, 1, 1)

l1 = plt.Line2D([0, 1], [0, 4], linewidth=1, linestyle="-", color="green")
l2 = plt.Line2D([0, 4.5], [0, 3], linewidth=1, linestyle="-", color="red")

ax.add_line(l1)
ax.add_line(l2)

angle = angle_plot(l1, l2, 0.25)
ax.add_patch(angle)

plt.show()

输出


相关文章