Jupyter笔记本以灰度显示matplotlib pcolor图像而不是颜色

时间:2017-04-15 20:13:42

标签: python matplotlib jupyter-notebook

我正在运行以下代码来显示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()

2 个答案:

答案 0 :(得分:1)

除了设置默认色图,您可以直接将其传递给pcolor方法:

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

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

答案 1 :(得分:0)

我发现这是因为默认的颜色图是'灰色'为了我。 plt.rcParams['image.cmap']

导致' Grays'。

我可以将它设置为' jet'现在显示颜色的图。

plt.rcParams['image.cmap'] = 'jet'
相关问题