Jupyter筆記本顯示matplotlib pcolor圖像的灰度而不是顏色

Question

我正在運行下面的代碼來顯示Z功能。輸出應該是彩色的。但是，它以灰度顯示。這是Jupyter筆記本。當我在其他地方跑步時，它顯示顏色。

# Import numpy and matplotlib.pyplot 
import numpy as np 
import matplotlib.pyplot as plt 

# Generate two 1-D arrays: u, v 
u = np.linspace(-2, 2, 41) 
v = np.linspace(-1,1,21) 

# Generate 2-D arrays from u and v: X, Y 
X,Y = np.meshgrid(u,v) 

# Compute Z based on X and Y 
Z = np.sin(3*np.sqrt(X**2 + Y**2)) 

# Display the resulting image with pcolor() 
plt.pcolor(Z) 
plt.show() 

Answer 1

我發現這是因爲默認的color-map對我來說是'Grays'。 plt.rcParams['image.cmap']

結果爲'灰色'。

我可以將它設置爲'jet'，它現在顯示顏色的情節。

plt.rcParams['image.cmap'] = 'jet' 

Answer 2

除了設置默認的顏色表，你可以直接傳遞一個到pcolor方法：

from matplotlib import cm 
plt.pcolor(Z, cmap = cm.viridis) 

在Jupyter筆記本電腦，你可以輸入cm.看到所有可用的顏色表後按TAB。