วิธีดึงข้อมูลจากกราฟ Matplotlib (Extract Plot Data)

หลังพลอตแล้ว สามารถดึงข้อมูล x, y กลับมาได้จาก Line2D object ด้วยเมธอด get_xdata() และ get_ydata() ใช้งานได้ทั้ง plot, scatter, และ artists อื่นๆ

ขั้นตอนการทำงาน

1. ตั้งค่าขนาดรูปและเลย์เอาต์
2. สร้างข้อมูล y และพลอต รับคืน curve object
3. ดึงข้อมูล: xdata = curve.get_xdata(), ydata = curve.get_ydata()
4. พิมพ์ผลลัพธ์และแสดงกราฟ

ตัวอย่างโค้ด

import numpy as np
from matplotlib import pyplot as plt

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

y = np.array([1, 3, 2, 5, 2, 3, 1])
curve, = plt.plot(y, c='red', lw=5)

print("Extracting data from plot....")
xdata = curve.get_xdata()
ydata = curve.get_ydata()
print("X data points:", xdata)
print("Y data points:", ydata)

plt.show()

ผลลัพธ์

Extracting data from plot....
X data points: [0. 1. 2. 3. 4. 5. 6.]
Y data points: [1 3 2 5 2 3 1]