python - 有条件地格式化 Python pandas 单元格

Question

我正在尝试根据单元格的值对 Python pandas DataFrame 进行着色、突出显示或更改。例如如果每行上的单元格大于该行第一列中的单元格，则将单元格突出显示为红色(或任何其他颜色)，否则保持原样。

我在这里写了一个for循环:

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:

You can apply conditional formatting, the visual styling of a DataFrame depending on the data within, by using the DataFrame.style property.

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)

---

编辑:要格式化特定单元格，您可以添加条件检查器以使用 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)