Условно формат Python pandas cell

Question

Я пытаюсь покрасить, выделить или изменить любовь Pandon pandas DataFrame на основе значения ячейки. например если ячейки в каждой строке больше, чем ячейка в первом столбце этой строки, выделите ячейку как красный (или любой другой цвет), иначе оставьте ее как есть.

Я написал цикл здесь:

for index in range(0, df.shape[0]): 
    for column in range(1, df.shape[1]): # from 1 not from 0 because I only need # to compare the 2nd to the last cell of each row with the 1st cell in the row 

     if df.iloc[index][column] - df_BDE_n_months_avg_std_pct.iloc[index][0] > 0: 
      then "PLEASE PUT YOUR HELP HERE, I NEED A PIECE OF CODE THAT CAN HIGHLIGHT THE CELL" 
     else: 
      "DO NOTHING" 

До сих пор я не нашел способ сделать это. Любая помощь будет отличной.

Answer 1

the style docs: От

Вы можете применить условное форматирование, визуальные стилизации в DataFrame в зависимости от данных в пределах, используя свойство DataFrame.style .

import pandas as pd 
df = pd.DataFrame([[2,3,1], [3,2,2], [2,4,4]], columns=list("ABC")) 

df.style.apply(lambda x: ["background: red" if v > x.iloc[0] else "" for v in x], axis = 1) 

---

Edit: форматировать специфические клетки, вы можете добавить условие шашек, чтобы проверить имя элемента с Series.iteritems() или проверить индекс с enumerate(), например, если вы хотите форматировать, начиная с колонки 3, вы можете использовать Перечислим и проверить индекс:

df = pd.DataFrame([[2,3,-3], [3,2,7], [2,4,4]], columns=list("ABC")) 

df.style.apply(lambda x: ["background-color: #ff33aa" 
          if (i >= 2 and (v > x.iloc[0] + x.iloc[1] 
              or v < x.iloc[0] - x.iloc[1])) 
          else "" for i, v in enumerate(x)], axis = 1)