Условное форматирование цвета в Pandas
Задача в условном форматировании (с использованием стиля, я думаю) Питон, панды
Есть табличка с двумя колонками
вторая таблица должна быть выделена условия:
- зеленым цветом, если номер 1-го столбца превышает номер второго;
- желтый, если номер 1-го столбца равен номеру второго;
- красным цветом, если номер 1-го столбца меньше второго.[
Спасибо за вашу помощь!
@jezrael ваша ссылка говорит об изображениях кода. Почему не изображения небольших фрагментов данных?
@SachinPrabhu - потому что та же проблема - невозможно скопировать данные
@jezrael Понял. Спасибо
Ответы 2
Вы можете применить форматирование столбца, используя pandas.DataFrame.style.apply и некоторые функции форматирования: https://pandas.pydata.org/pandas-docs/stable/user_guide/style.html
Пожалуйста, добавьте сюда некоторые функции форматирования на случай, если ссылка когда-нибудь сломается.
Идея заключается в том, чтобы создать новый DataFrame, заполненный стилями по условию с Styler.apply, для набора строк по условиям используется DataFrame.mask:
def highlight(x):
c1 = 'background-color: green'
c2 = 'background-color: yellow'
c3 = 'background-color: red'
m1 = x.iloc[:, 0] > x.iloc[:, 1]
m2 = x.iloc[:, 0] == x.iloc[:, 1]
df1 = pd.DataFrame(c3, index=x.index, columns=x.columns)
return df1.mask(m1, c1).mask(m2, c2)
df.style.apply(highlight, axis=None)
Обновлено:
Если нужно установить только один столбец, используйте numpy.select:
def highlight(x):
c1 = 'background-color: green'
c2 = 'background-color: yellow'
c3 = 'background-color: red'
c = ''
m1 = x.iloc[:, 0] > x.iloc[:, 1]
m2 = x.iloc[:, 0] == x.iloc[:, 1]
df1 = pd.DataFrame(c, index=x.index, columns=x.columns)
df1.iloc[:, 1] = np.select([m1, m2], [c1, c2], default=c3)
return df1