python - 将brewer2mpl发散色图与matplotlib一起使用时，gamma值不等于1的结果差

在this question中使用颜色图在注释中被介绍给brewer2mpl。对于发散图（中间为零），我注意到伽玛以一种幼稚的方式实现，因此当它不等于1时会给出意外的†结果。中性色调表示零滑动，因此值接近零是伽玛= 0.5的蓝色和伽玛= 1.5的猩猩

无论如何，是否可以使类似的参数在matplotlib或brewer2mpl中围绕零对称地运行，还是我只需要通过将np.abs(u)提高到一定的幂然后重新归一化并重新引入标志？

†预期行为是将伽玛对称地应用于零附近；至少那是我所期望的。

import numpy as np
import matplotlib.pyplot as plt
import brewer2mpl

Re = 6378137. # meters
J2 = 1.7555E+25 # m^5/s^2
hw = 400
x = np.linspace(-2.5*Re, 2.5*Re, 2*hw+1)
x, z = np.meshgrid(x, x)
r = np.sqrt(x**2 + z**2)
u = J2 * r**-5 * 0.5 * (3*z**2 - r**2)
u[r<Re] = np.nan

if True:
    umax = np.nanmax(np.abs(u))
    bmap = brewer2mpl.get_map('RdBu', 'Diverging', 9)
    gammas = 0.5, 1, 1.5
    plt.figure()
    for i, gamma in enumerate(gammas):
        cmap = bmap.get_mpl_colormap(N=100, gamma=gamma)
        plt.subplot(1, 3, i+1)
        plt.imshow(u, cmap=cmap, vmin=-umax, vmax=umax)
        plt.colorbar()
        plt.title('gamma = ' + str(gamma))
        plt.plot([hw, hw], [0.3*hw, 1.7*hw], '-k')
        plt.xlim(0, 2*hw+1)
        plt.ylim(0, 2*hw+1)
    plt.suptitle("Earth's geopotential's J2 component", fontsize=16)
    plt.show()

最佳答案

您可以创建两个颜色图，一个从红色到白色，一个从白色到蓝色。然后将伽玛应用于每个。最后，从这些颜色图中获取颜色，并使用组合的颜色创建一个新颜色。

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import LinearSegmentedColormap
from copy import copy

cmap = plt.get_cmap("RdBu", lut=256)
# Create two new colormaps, each with one half of the original
cmap_lower = LinearSegmentedColormap.from_list("", cmap(np.arange(0,128)))
cmap_upper = LinearSegmentedColormap.from_list("", cmap(np.arange(128,256)))
gammas = [1, 0.5, 1.5]


fig, axs = plt.subplots(ncols=3, figsize=(8,5))

for ax, gamma in zip(axs, gammas):
    # copy each colormap and set the respective gamma
    cm1 = copy(cmap_lower)
    cm1.set_gamma(gamma)
    cm2 = copy(cmap_upper)
    cm2.set_gamma(gamma)
    # get the colors from the each
    colors = np.concatenate((cm1(np.arange(0,256)), cm2(np.arange(0,256))), axis=0)
    this_cmap = LinearSegmentedColormap.from_list("", colors)

    im = ax.imshow(np.arange(300).reshape(30,10), cmap=this_cmap)
    fig.colorbar(im, ax=ax)
    ax.set_title(f"gamma={gamma}")

plt.show()

关于python - 将brewer2mpl发散色图与matplotlib一起使用时，gamma值不等于1的结果差，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/59931310/