matplotlib的色彩映射

问题描述

我是python的新手，安装完成后，我使用matplotlib绘制了我的3D数据。遗憾的是，我不知道如何做的唯一的事情是颜色部分。我的图像只是显示表面，但不使用颜色栏。这里是我的代码。

I'm new to python and after installing it I've accomplished to plot my 3d data using matplotlib. Sadly the only thing I don't know how to get done is the color part. My image just shows the surface but doesn't use the color bar at all. Here is my code.

from mpl_toolkits.mplot3d import Axes3D
from matplotlib import cm
from matplotlib.ticker import LinearLocator, FormatStrFormatter
import matplotlib.pyplot as plt
from matplotlib.mlab import griddata
import numpy as np

fig = plt.figure()
ax = fig.gca(projection='3d')

data = np.genfromtxt('Uizq.txt')
x = data[:,0]
y = data[:,1]
z = data[:,2]

xi = np.linspace(min(x), max(x))
yi = np.linspace(min(y), max(y))

ax.set_xlabel('X')
ax.set_ylabel('Y')
ax.set_zlabel('U')

X, Y = np.meshgrid(xi, yi)
Z = griddata(x, y, z, xi, yi)
ax.set_zlim3d(np.min(Z), np.max(Z))


surf = ax.plot_surface(X, Y, Z, rstride=2, cstride=2, cmap=cm.jet,
                   linewidth=0.5, antialiased=False)

fig.colorbar(surf, shrink=0.5, aspect=5)

plt.show()

你显然可以看到它是蓝色的，我想使用完整的 cm.jet 将颜色与U光谱。这可能是一个非常noob的问题，所以很抱歉，如果你滚过你的眼睛。

you can obviously see that it is all blue, and I want to relate the color with "U" using the full cm.jet spectrum. This might be a very noob question, so sorry if you rolled your eyes.

推荐答案

添加行

surf.set_clim([np.min(Z),np.max(Z)])

在添加颜色栏之前。

似乎3D绘图没有考虑掩蔽，因此您在数据中包含 NaN ，这会混淆自动颜色限制。

It seems that the 3D plotting does not take into account the masking, so you are including NaN in the data, which confuses the automatic color limits.

这篇关于matplotlib的色彩映射的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持！