Если число ниже, то cv2.putText красного цвета, если выше, то зеленого, как?

Question

[JRBRO]

Есть оценка изображения, которая выводит оценку в таком стиле "40.012353452342457845"

score1 = brisque.score(img1)
        print("Score for"+(b)+" ", score1)
        cv2.putText(
        img = img,
        text = str(score1),
        org = (0, 50),
        fontFace = cv2.FONT_HERSHEY_DUPLEX,
        fontScale = 2.0,
        color = (125, 246, 55),
        thickness = 3
        )

Как максимально правильно написать логику, чтобы цвет текста зависел от числа?
Я понимаю, что это логика if/else, но единственное что я хочу менять это цвет, остальные параметры остаются как есть. Как такое реализовать, чтобы не делать 100 лишних строк?

Answer 1

[Николай Савельев]

if ... else

Comments

[JRBRO]

Я понимаю что это if/else, не могу понять куда его конкретно всунуть.
Предполагал вариант

if score >10
        cv2.putText(
        img = img,
        text = str(score1),
        org = (0, 50),
        fontFace = cv2.FONT_HERSHEY_DUPLEX,
        fontScale = 2.0,
        color = (2, 2, 2),
        thickness = 3
        )
else
        cv2.putText(
        img = img,
        text = str(score1),
        org = (0, 50),
        fontFace = cv2.FONT_HERSHEY_DUPLEX,
        fontScale = 2.0,
        color = (1, 1, 1),
        thickness = 3
        )

но хотелось бы покороче

[U235U235]

JRBRO, без if:

cv2.putText(
        img = img,
        text = str(score1),
        org = (0, 50),
        fontFace = cv2.FONT_HERSHEY_DUPLEX,
        fontScale = 2.0,
        color = ((1, 1, 1),(2,2,2))[score >10]
        thickness = 3
        )

[Vindicar]

JRBRO, в питоне есть тернарные выражения, хотя они и громоздкие =(
a = 'foo' if x > 0 else 'bar'
Это то же самое что и

if x > 0:
    a = 'foo' 
else:
    a = 'bar'

[JRBRO]

U235U235, Vindicar, Спасибо большое, оба варианта супер!

Answer 2

[Василий]

text_color = (1, 1, 1) #по умолчанию текст красный

if score >10:
    text_color = (2, 2, 2) #если оценка оказалась выше 10, то меняем цвет
if score > 50:
    text_color = (3, 3, 3) #можно увеличить количество цветов, в зависимости от оценки

cv2.putText(
img = img,
text = str(score1),
org = (0, 50),
fontFace = cv2.FONT_HERSHEY_DUPLEX,
fontScale = 2.0,
color = text_color,
thickness = 3
)

Answer 3

[Сергей Водаков]

Есть такая штуковина, называется переменные. Отлично подходит в данном случае!
Заводим переменную, ну скажем text_color и присваиваем ей цвет, по условию, в зависимости от score1, а потом вместо: color = (125, 246, 55) пишем color = text_color