指定 Matplotlib 中的色彩對映的中間點


使用 plt.subplots(1, 1) 方法,我們可以建立影像和座標軸。我們可以使用 fig.colorbar 在圖片的中間位置製作顏色條。

步驟

  • 使用 mgrid() 方法,`nd_grid` 例項返回一個開放的多維“網格”。

  • 建立 Z1、Z2 和 Z 資料。

  • 使用子圖方法建立 fig 和 ax 變數,其中,預設的 nrows 和 ncols 為 1,使用子圖() 方法。

  • 使用 colorbar() 方法為 ScalarMappable 例項 *mappable* 建立一個顏色條。

  • 使用 plt.show(),我們可以顯示影像。

例項

import numpy as np
import matplotlib.pyplot as plt
import matplotlib.colors as colors

N = 100
X, Y = np.mgrid[-3:3:complex(0, N), -2:2:complex(0, N)]

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

fig, ax = plt.subplots(1, 1)

fig.colorbar(ax.pcolor(X, Y, Z, norm=colors.LogNorm(vmin=Z.min(),
vmax=Z.max()), shading='auto'))
plt.show()

輸出

更新於: 16-3-2021

319 次瀏覽

啟動您的職業生涯

完成課程認證

開始
廣告
© . All rights reserved.