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

วิธีพลอตแผนที่ความหนาแน่น (Density Map) ด้วย Python Matplotlib

เพื่อสร้างแผนที่ความหนาแน่น (density map) ใน Python สามารถใช้ pcolormesh() ร่วมกับข้อมูลจากฟังก์ชันเอ็กซ์โพเนนเชียลได้

ขั้นตอน

  • สร้างข้อมูล side, x, y, z ด้วย NumPy (linspace ช่วยสร้างข้อมูลระหว่างสองจุด)
  • สร้างเมทริกซ์พิกัด X, Y ด้วย meshgrid จากข้อมูล side
  • คำนวณค่า Z แบบเอ็กซ์โพเนนเชียลจาก X และ Y
  • พลอต pseudocolor plot ด้วย pcolormesh() พร้อม shading='auto'
  • แสดงกราฟด้วย show()

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

from matplotlib import pyplot as plt, cm, colors
import numpy as np

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

side = np.linspace(-2, 2, 15)
X, Y = np.meshgrid(side, side)
Z = np.exp(-((X - 1) ** 2 + Y ** 2))

plt.pcolormesh(X, Y, Z, shading='auto')
plt.show()

ผลลัพธ์

วิธีพลอตแผนที่ความหนาแน่น (Density Map) ด้วย Python Matplotlib