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

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

从样式文档：

您可以通过使用 DataFrame.style 属性应用条件格式，即 DataFrame 的视觉样式，具体取决于其中的数据。

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)

Answer 2

import numpy as np
import pandas as pd
df = pd.DataFrame(np.random.rand(4,3))
df.style.applymap(lambda x: 'background-color : yellow' if x>df.iloc[0,0] else '')