如何在绘制后修改散点图的颜色图

Question

例如，您可以使用scatter(X, Y, c=..., cmap="rainbow") 将颜色图分配给散点图，但是以后有没有办法修改颜色图？

我知道我们可以通过set_color() 和set_alpha() 更改透明度和颜色，但我没有找到颜色映射的等价物。一种解决方法是保存散点图的数据，将其擦除，然后再次使用新参数scatter()，但这会很昂贵。这是一个问题，因为这个问题的目标是能够实时更改c 参数，以便旧点“冷却”到更深的颜色。

感谢您的宝贵时间。

---

编辑以添加更多信息：我希望能够在使用单一颜色的散点图和具有可变颜色图和标准化的散点图之间来回切换。例如，在scatter(X, Y, color="#FF0000") 和scatter(X, Y, c=age, cmap="rainbow") 之间，带有一个变量age = [i for i,x in enumerate(X)]。

Answer 1

正如@ImportanceOfBeingErnest 所指出的，使用set_cmap() 似乎相当简单：

x = np.linspace(0,10,10)
y = x**2
c = np.linspace(-10,10,10)
cmap = "seismic"
norm = matplotlib.colors.Normalize(vmin=min(c),vmax=max(c))



from matplotlib.widgets import RadioButtons

fig, ax = plt.subplots()

scat = plt.scatter(x,y,c=c,cmap=cmap,norm=norm)
plt.colorbar()
scat.update_scalarmappable()
scat.set_facecolors('k')

plt.subplots_adjust(left=0.3)

axcolor = 'lightgoldenrodyellow'
rax = plt.axes([0.05, 0.7, 0.15, 0.15], facecolor=axcolor)
radio = RadioButtons(rax, ('no-cmap','cmap'))

def colorfunc(label):
    if label=="cmap":
        scat.set_cmap(cmap)
        scat.set_norm(norm)
    elif label=="no-cmap":
        scat.set_facecolors('k')
    else:
        raise KeyError("invalid label")
    plt.draw()
radio.on_clicked(colorfunc)

Answer 2

您实际上可以更改值，而不是更改颜色图。如果您为颜色图设置了under 颜色，则可以将值更改为低于标准的任何值并使其以该颜色显示。

当按下 n 时，以下会将值设置为比最小值小一。按下 m 后，它们会被设置回原来的值。

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(42)

a = np.random.randint(10,30, size=20)
x = np.random.rand(20)
y = np.random.rand(20)

cmap = plt.get_cmap("rainbow")
cmap.set_under("black")

fig, ax = plt.subplots()
ax.set_title("Press n and m keys to change colors")
sc = ax.scatter(x,y, c=a, cmap=cmap)

def change(evt):
    if evt.key in list("nm"):
        if evt.key == "n":
            sc.set_array(np.ones_like(a)*a.min()-1)
        if evt.key == "m":
            sc.set_array(a)
        fig.canvas.draw_idle()


fig.canvas.mpl_connect("key_press_event", change)

plt.show()

---

如果您确实需要从没有定义其数组的散点图开始，您可以如下所示进行。

import matplotlib.pyplot as plt
import numpy as np; np.random.seed(42)

a = np.random.randint(10,30, size=20)
x = np.random.rand(20)
y = np.random.rand(20)

cmap = plt.get_cmap("rainbow")
cmap.set_under("black")
norm = plt.Normalize(a.min(), a.max())

fig, ax = plt.subplots()
ax.set_title("Press n and m keys to change colors")
sc = ax.scatter(x,y, color="black")

def change(evt):
    if evt.key in list("nm"):
        sc.set_cmap(cmap)
        sc.set_norm(norm)
        if evt.key == "n":
            sc.set_array(np.ones_like(a)*a.min()-1)
        if evt.key == "m":
            sc.set_array(a)
        fig.canvas.draw_idle()
        

fig.canvas.mpl_connect("key_press_event", change)

plt.show()