如何在兩個不同的圖上將plt.imshow（）的顏色圖設置為相同的值

Question

考慮以下。 我有兩個情節：

fig = plt.figure()
plt.imshow(nonfour, cmap='gray')

fig = plt.figure()
plt.imshow(np.abs(four), cmap='gray')

這兩個圖是使用數組nonfour和np.abs（four）創建的。 創建每個圖時，使用每個數組的最小值和最大值設置顏色圖。

我想做的是從第一個圖生成一個cmap並將其用作第二個圖的cmap

（即第二個圖的顏色圖將基於我的第一個圖的最大值和最小值）。

Answer 1

正如imshow指出的， imshow使用關鍵字vmin和vmax來控制顏色圖的比例。 您可以使用get_clim()方法獲取第一個圖的（自動確定的）值：

# Create your first plot
img = plt.imshow(nonfour, cmap='gray')

# Extract vmin and vmax
vmin, vmax = img.get_clim()

# Create your second plot using these limits
plt.imshow(np.abs(four), cmap='gray', vmin=vmin, vmax=vmax)

另外，您也可以使用set_clim()方法：

img = plt.imshow(nonfour, cmap='gray')
img2 = plt.imshow(np.abs(four), cmap='gray')
img2.set_clim(img.get_clim())