python - 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

除了设置默认颜色图之外，您还可以将颜色图直接传递给 pcolor 方法:

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

在 Jupyter 笔记本中，您可以在输入 cm. 后按 TAB 以查看所有可用的颜色图。