python - 使用 Dataframe 单元格值设置另一个具有相同尺寸的 Dataframe 的样式

Question

import pandas as pd
import numpy as np
df = pd.DataFrame({'A': 'foo bar foo'.split(),
               'B': 'one one two'.split(),
               'C': np.arange(3), 'D': np.arange(3) * 2})

j = [{'bgcolor': "#55aa2a"}, {'bgcolor': "#d42a2a"}, {'bgcolor': "#d42a2a"}]
df2 = pd.DataFrame({'E': j, 'F': j, 'G': j, 'H': j})

上面的代码产生了两个数据帧，df1 是一个标准帧，df2 是由字典组成的帧(每个单元格都有一个字典作为它的值)。

我想使用 df2 的单元格来设置 df 单元格的样式，即单元格 df[0,1]将采用单元格 df2[0,1] 的值并将其用作其样式

示例:

def highlight(df,df2):
    df[0,1] = '{}'.format(df2[0,1])
    return df

(应用于整个框架除外)

这应该将 df[0,1] 的背景颜色设置为 df2[0,1]="55aa2a" 但返回一个 KeyError 在调用 df = df.style.apply(highlight, df2=df2).render()

之后

是否可以使用 df2 的单元格来设置 df1 的单元格样式？

Answer 1

您可以更改字符串值的格式，然后返回具有相同列名的 DataFrame(索引值也必须相同):

df2 = df2.applymap(lambda x: 'background-color: {}'.format(x.get('bgcolor')))
print (df2)
                           E                          F  \
0  background-color: #55aa2a  background-color: #55aa2a   
1  background-color: #d42a2a  background-color: #d42a2a   
2  background-color: #d42a2a  background-color: #d42a2a   

                           G                          H  
0  background-color: #55aa2a  background-color: #55aa2a  
1  background-color: #d42a2a  background-color: #d42a2a  
2  background-color: #d42a2a  background-color: #d42a2a  

def highlight(x):
    d = dict(zip(df2.columns, x.columns))
    return df2.rename(columns=d)

或者:

def highlight(x):
    return pd.DataFrame(df2.values, columns=x.columns)

---

df.style.apply(highlight, axis=None)