如何在 Matplotlib 中繪製等值線圖和對數色標？

要在 Matplotlib 中繪製等值線圖和對數色標，我們可以採取以下步驟 −

• 設定圖片大小並調整子圖之間和周圍的填充。
• 初始化一個變數，N，用於樣本資料的數量。
• 使用 numpy 建立x, y, X, Y, Z1, Z2 和z 資料點。
• 建立一個 figure 和一組子圖。
• 使用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()

輸出

Rishikesh Kumar Rishi

更新於: 15-06-2021

2 千次瀏覽

開啟你的 事業

完成課程即可獲得認證

開始