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

在 Matplotlib 中，绘制两条线之间角度的最佳方法是使用Arc类创建角度弧来绘制之间的角度。

步骤

  • 设置图形大小并调整子图之间和周围的填充。
  • 使用figure()方法创建一个新图形或激活一个现有图形。
  • 使用add_subplot()方法将'~.axes.Axes'作为子图布局的一部分添加到图形中。
  • 将二维线实例创建为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()

输出

更新于:03-Jun-2021

2K+ 浏览

开启您的 职业

通过完成该课程获得认证

开始
广告