color/colormap.py


1
2
3
4
5
6
7
8
9
10
11
12
13

import numpy as np
import matplotlib.cm as cm

def map_to_color(a, range=None, map=cm.hsv):
    a = np.asarray(a)
    if range is None:
        range = (a.min(), a.max())

    ax = (a - range[0])/(range[1]-range[0])

    rgba = (map(ax)*255).astype(np.uint32)

    return rgba[:,0] << 16 | rgba[:,1] << 8 | rgba[:,2]