Computer >> คอมพิวเตอร์ >  >> การเขียนโปรแกรม >> Python

วิธีที่ดีที่สุดในการพล็อตมุมระหว่างสองบรรทัดใน Matplotlib


วิธีที่ดีที่สุดในการพล็อตมุมระหว่างสองบรรทัดใน Matplotlib คือการใช้ Arc class เพื่อสร้างส่วนโค้งของมุมเพื่อพล็อตมุมระหว่าง

ขั้นตอน

  • กำหนดขนาดรูปและปรับช่องว่างภายในระหว่างและรอบๆ แผนผังย่อย
  • สร้างตัวเลขใหม่หรือเปิดใช้งานตัวเลขที่มีอยู่โดยใช้ figure() วิธีการ
  • เพิ่ม '~.axes.Axes' ไปยังรูปที่เป็นส่วนหนึ่งของการจัดเรียงแผนย่อยโดยใช้ add_subplot() วิธีการ
  • สร้างอินสแตนซ์เส้น 2D เป็น l1 และล2 .
  • เพิ่มเส้นให้กับแกนปัจจุบัน
  • ในการพล็อตมุม ให้เรียกใช้เมธอดที่ผู้ใช้กำหนดซึ่งส่งคืนส่วนโค้งของวงรี ความยาวของส่วนโค้งสามารถสร้างขึ้นได้โดยใช้ความชันของเส้น..
  • เพิ่มศิลปิน เช่น ส่วนโค้ง โดยใช้ add_patch() วิธีการ
  • หากต้องการแสดงรูป ให้ใช้ 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()

ผลลัพธ์

วิธีที่ดีที่สุดในการพล็อตมุมระหว่างสองบรรทัดใน Matplotlib