matplotlib - 如何离散化内置颜色图？

Question

我想使用“地形”颜色图绘制高程图。但是，我想通过我选择的边界对其进行离散化。例如：

bounds = [0, 500, 1000, 1500, 3000, data.max()]

有没有办法离散化现成的颜色图？

标签： matplotlibcolors

Answer 1

试试这个，它将以 N 种颜色离散化颜色图，并在您绘制时将其设置为默认颜色循环。

from matplotlib import rcParams, cycler

N = 6
cmap = plt.cm.terrain
custom_cycler = cycler(color=cmap(np.linspace(0, 1, N)))
ax.set_prop_cycle(custom_cycler)