计算一系列值的RGB值以创建热图

Question

我试图用python创建一个热图。为此，我必须为可能值范围内的每个值指定一个RGB值。我想把颜色从蓝色（最小值）改为绿色（最大值）。

下面的图片示例解释了我对颜色组成的看法：我们有1（纯蓝色）到3（纯红色）的范围，2介于绿色之间。

我读了关于线性插值的文章，写了一个函数（或多或少）处理最小值和最大值之间某个值的计算，并返回一个RGB元组。它使用if和elif条件（这并不能让我完全满意）：

def convert_to_rgb(minimum, maximum, value):
    minimum, maximum = float(minimum), float(maximum)    
    halfmax = (minimum + maximum) / 2
    if minimum <= value <= halfmax:
        r = 0
        g = int( 255./(halfmax - minimum) * (value - minimum))
        b = int( 255. + -255./(halfmax - minimum)  * (value - minimum))
        return (r,g,b)    
    elif halfmax < value <= maximum:
        r = int( 255./(maximum - halfmax) * (value - halfmax))
        g = int( 255. + -255./(maximum - halfmax)  * (value - halfmax))
        b = 0
        return (r,g,b)

不过，我想知道是否可以在不使用if条件的情况下为每个颜色值编写一个函数。有人知道吗？非常感谢！

Answer 1

def rgb(minimum, maximum, value):
    minimum, maximum = float(minimum), float(maximum)
    ratio = 2 * (value-minimum) / (maximum - minimum)
    b = int(max(0, 255*(1 - ratio)))
    r = int(max(0, 255*(ratio - 1)))
    g = 255 - b - r
    return r, g, b