熊猫-使用混合类型的列为数据框设置样式

Question

你好朋友， 我想在包含数字和非数字列类型的数据框中应用样式功能。 如何避免出现以下错误？

---

错误： TypeError ：（“ str”和“ int”的实例之间不支持“”>”，“发生在索引A”）

---

---

import pandas as pd
d = {'A': ['A','B','C']}
data = pd.DataFrame(data=d)
data['B']=[7,8,9]
data['C']=[1,2,3]

def color_negative_red(val):
    """
    Takes a scalar and returns a string with
    the css property `'color: red'` for negative
    strings, black otherwise.
    """
    color = 'white' if (val > 2) else 'black'
    return 'color: %s' % color

df_new = data.style.applymap(color_negative_red)

---

Answer 1

如果要创建自定义函数，则是一个更复杂的数据的想法-仅选择带有DataFrame.select_dtypes的数字列，将输出df1设置为空字符串，并用numpy.where替换数字值：

def color_negative_red(x):
    c1 = 'background-color: red'
    c2 = 'background-color: black' 

    cols = x.select_dtypes(np.number).columns
    df1 = pd.DataFrame('', index=x.index, columns=x.columns)
    df1[cols] = np.where(x[cols] > 2, c1, c2)
    return df1

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