Exemplo n.º 1
0
def joincmap(cmap1, cmap2, N=256):
    """
    Join two colormaps and return one dynamic colormap.
    """
    if isinstance(cmap1, str):
        cmap1 = plt.get_cmap(cmap1)

    if isinstance(cmap2, str):
        cmap2 = plt.get_cmap(cmap2)

    cmap = ListedColormap(
        np.vstack((cmap1(np.linspace(0, 1,
                                     N // 2)), cmap2(np.linspace(0, 1,
                                                                 N // 2)))),
        '->'.join((cmap1.name, cmap2.name)), N)

    cmap._init()

    cmap.values = np.linspace(-1, 1, N)
    cmap.hinge = 0
    return DynamicColormap(cmap)