quickmaps.py 文件源码

def make_object_map(data,field,**kwargs):
    linear = False
    for key,value in kwargs.iteritems():
        if key == 'linear':
            linear = value
    print linear
    if linear == False:
        colors,rangelist = make_distributed_range(data,field)
    else:
        colors = get_heatmap51()
        colors2 = colors 
        maxvalue = data[field].max()
        if maxvalue < 51:
            totallist = range(maxvalue)
            colors = reduce_color_list_size(totallist,colors)
            colors,rangelist = make_gradient_range(data[field].min(),maxvalue,colors)
        else:
            colors = reduce_color_list_size(range(len(data)),colors)
            colors,rangelist = make_gradient_range(data[field].min(),maxvalue,colors)
            if not rangelist[0] == 0:
                rangelist = [0] + rangelist[1:]
            data['COLORKEY'] = pd.cut(data[field],bins=rangelist+[1000000000],labels=colors)

            return data
    colors2 = get_heatmap51()
    if not rangelist[0] == 0:
        rangelist = [0] + rangelist[1:]
    data['COLORKEY'] = pd.cut(data[field],bins=rangelist,labels=colors[1:])

    return data

# for a given dataframe and field returns a non used grouped object to multiple operations on