ฉันจะวาดเส้นแนวโน้มกระจายโดยใช้ Matplotlib ได้อย่างไร

ในการวาดเส้นแนวโน้มกระจายโดยใช้ matplotlib เราสามารถใช้ polyfit() และ poly1d() วิธีการรับจุดเส้นแนวโน้ม

ขั้นตอน

• กำหนดขนาดรูปและปรับช่องว่างภายในระหว่างและรอบๆ แผนผังย่อย

• สร้างจุดข้อมูล x และ y โดยใช้ numpy

• สร้างร่างและชุดแผนย่อย

• พล็อตจุดข้อมูล x และ y โดยใช้ numpy

• ค้นหาจุดข้อมูลเส้นแนวโน้มโดยใช้ polyfit() และ poly1d() วิธีการ

• พล็อต x และ p(x) จุดข้อมูลโดยใช้ plot() วิธีการ

• หากต้องการแสดงรูป ให้ใช้ show() วิธีการ

ตัวอย่าง

import numpy as np
from matplotlib import pyplot as plt

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

x = np.random.rand(100)
y = np.random.rand(100)
fig, ax = plt.subplots()
_ = ax.scatter(x, y, c=x, cmap='plasma')
z = np.polyfit(x, y, 1)
p = np.poly1d(z)
plt.plot(x, p(x), "r-o")

plt.show()

ผลลัพธ์