循环移位色彩图

Question

有没有办法循环移动hsv色彩地图可以改变中心色彩吗？它在我的身材中占据了一些空间，我希望它是一种不同的颜色。由于色彩映射是周期性的，这应该是可能的。

Answer 1

基于本example中使用的非线性颜色表，但具有线性水平由移位取代，

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

class roll_cmap(LinearSegmentedColormap): 

    def __init__(self, cmap, shift): 

     assert 0. < shift < 1. 
     self.cmap = cmap 
     self.N = cmap.N 
     self.monochrome = self.cmap.monochrome 
     self._x = np.linspace(0.0, 1.0, 255) 
     self._y = np.roll(np.linspace(0.0, 1.0, 255),int(255.*shift)) 

    def __call__(self, xi, alpha=1.0, **kw): 
     yi = np.interp(xi, self._x, self._y) 
     return self.cmap(yi, alpha) 


if __name__ == '__main__': 

    y, x = np.mgrid[0.0:3.0:100j, 0.0:5.0:100j] 
    H = np.sin(8*x/np.pi) 

    cmap = plt.cm.hsv 
    cmap_rolled = roll_cmap(cmap, shift=0.8) 

    plt.subplot(2,1,1) 
    plt.contourf(x, y, H, cmap=cmap) 
    plt.colorbar() 
    plt.subplot(2,1,2) 
    plt.contourf(x, y, H, cmap=cmap_rolled) 
    plt.colorbar() 

    plt.show() 

这导致下面的输出，