python - 如何获得用于 4D 绘图 Matplotlib 的颜色图中的色阶

Question

我有一个 4D 情节：

import numpy as np
from mpl_toolkits.mplot3d import Axes3D
import matplotlib.pyplot as plt
from matplotlib import cm

x = np.arange(100)/ 101
y = np.sin(x) + np.cos(x)
X,Y = np.meshgrid(x,y)
Z = (X**2) / (Y**2)
A = np.sin(Z)

fig = plt.figure()
ax = fig.add_subplot(111,projection= '3d' )
ax.plot_surface(X,Y, Z, facecolors=cm.Oranges(A))
plt.show()

输出图像

但是如何查看颜色图的比例？

Answer 1

您需要从颜色图创建一个可映射的

m = cm.ScalarMappable(cmap=cm.Oranges)
m.set_array(A)
cbar = plt.colorbar(m)

正如对这个问题的回答： Matplotlib 3d plot - Associate colorbar with different axis