包含字符串 Pandas 的单元格中的颜色

Question

如果列包含特定字符串，我正在尝试更改数据框中的背景颜色，但我似乎无法让它工作。到目前为止我有以下代码：

def highlight_cells(StartingDataFrame):
    if StartingDataFrame[StartingDataFrame['HRC'].str.contains("HRC")]:
        return ['background-color: red']*5
    else:
        return ['background-color: white']*5

StartingDataFrame.style.apply(highlight_cells, axis=1)

但它似乎对细胞没有任何作用。我做错了什么吗？

代码：

StartingDataFrame = pd.DataFrame({'HRC':['aaa','HRD ','HRC oo'],
                              'A':[1,2,3]})


def highlight_cells(x): 
   c1 = 'background-color: red'
   c = 'background-color: white'

   #if True are strings
   m1 = StartingDataFrame['HRC'].str.contains("HRC")

   df1 = pd.DataFrame(c, index=x.index, columns=x.columns)
   df1.loc[m1, 'HRC'] = c1
   return df1
StartingDataFrame.style.apply(highlight_cells,axis=None)
StartingDataFrame.to_excel("outputTest.xlsx")

Answer 1

您可以使用自定义函数来创建样式的 DataFrame：

StartingDataFrame = pd.DataFrame({'HRC':['aaa','HRD ','HRC oo'],
                                  'A':[1,2,3]})


def highlight_cells(x): 
   c1 = 'background-color: red'
   c = 'background-color: white'

   #if True are strings
   m1 = StartingDataFrame['HRC'].str.contains("HRC", na=False)

   df1 = pd.DataFrame(c, index=x.index, columns=x.columns)
   df1.loc[m1, 'HRC'] = c1
   return df1

(StartingDataFrame.style.apply(highlight_cells,axis=None)
                  .to_excel("outputTest.xlsx", index=False))

另一个解决方案是通过 in:

选择测试值的列

def highlight_cells(val):
    color = 'red' if 'HRC' in val else 'white'
    return f'background-color: {color}' 

(StartingDataFrame.style.applymap(highlight_cells, subset=['HRC'])
                  .to_excel("outputTest.xlsx", index=False))

编辑：在你的解决方案中需要分配回来：

styles = StartingDataFrame.style.apply(highlight_cells,axis=None)
styles.to_excel("outputTest.xlsx")

包含字符串 Pandas 的单元格中的颜色

Color in cell containing a string Pandas

python

colors

background-color

dataframe

pandas