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

จะพล็อตคอนทัวร์และบันทึกมาตราส่วนสีใน Matplotlib ได้อย่างไร


ในการพล็อตคอนทัวร์และสเกลบันทึกใน Matplotlib เราสามารถทำตามขั้นตอนต่อไปนี้ -

  • กำหนดขนาดรูปและปรับช่องว่างภายในระหว่างและรอบๆ แผนผังย่อย
  • เริ่มต้นตัวแปรN สำหรับจำนวนข้อมูลตัวอย่าง
  • สร้าง x, y, X, Y, Z1, Z2 และ z จุดข้อมูลโดยใช้ตัวเลข
  • สร้างร่างและชุดแผนย่อย
  • พล็อตรูปร่างโดยใช้ contourf() วิธีการ
  • สร้างแถบสีสำหรับอินสแตนซ์ที่แมปแบบสเกลาร์ได้
  • หากต้องการแสดงรูป ให้ใช้ show() วิธีการ

ตัวอย่าง

import matplotlib.pyplot as plt
import numpy as np
from numpy import ma
from matplotlib import ticker, cm

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

N = 100
x = np.linspace(-3.0, 3.0, N)
y = np.linspace(-2.0, 2.0, N)

X, Y = np.meshgrid(x, y)

Z1 = np.exp(-X**2 - Y**2)
Z2 = np.exp(-(X * 10)**2 - (Y * 10)**2)

z = Z1 + 50 * Z2
z[:5, :5] = -1
z = ma.masked_where(z <= 0, z)
fig, ax = plt.subplots()

cs = ax.contourf(X, Y, z, locator=ticker.LogLocator(), cmap=cm.PuBu_r)
cbar = fig.colorbar(cs)

plt.show()

ผลลัพธ์

จะพล็อตคอนทัวร์และบันทึกมาตราส่วนสีใน Matplotlib ได้อย่างไร