Agrupar colores en cajas(operadores)

def color_groups(color):
    if color == 'red':
        return 'warm'
    elif color == 'blue':
        return 'cold'
    else:
        return 'neutral'
    
color_groups('blue')