我正在尝试打印具有背景渐变的熊猫数据框,以提高可读性。我尝试将在docs中找到的内容应用于一个简单的用例,但是我无法让jupyter笔记本实际用颜色打印表格-我一直在获取纯数据框。 小例子:
import seaborn as sns
import pandas as pd
cm = sns.light_palette('green', as_cmap=True)
df_res = pd.DataFrame(index =['foo','bar'],columns = ['Value 1','Value 2','Value 3'])
df_res.loc['foo'] = [-.5*100, .3,.2]
df_res.loc['bar'] = [.3*100, .6,.9]
df_res.style.background_gradient(cmap=cm)
仅打印
我尝试了不同的打印技术,即
pretty = df_res.style.background_gradient(cmap=cm)
display(pretty)
或
print(pretty)
或其他颜色图
df_res.style.background_gradient(cmap='viridis')
但是它们都不起作用。我还尝试了样式器是否可以正常工作,但至少applymap函数可以实现应有的功能:
def color_negative_red(val):
"""
Takes a scalar and returns a string with
the css property `'color: red'` for negative
strings, black otherwise.
"""
color = 'red' if val < 0 else 'black'
return 'color: %s' % color
df_res.style.applymap(color_negative_red)
可打印
所以不确定为什么background_gradient似乎没有效果。
编辑:刚刚找到了原因。这是一个简单的解决方法,但是如果其他人遇到同样的问题,我会继续努力。 显然,大熊猫使用元素而不是浮点数来初始化数据框。如此简单地将初始化更改为
df_res = pd.DataFrame(index =['foo','bar'],columns = ['Value 1','Value 2','Value 3']).astype('float')
解决了这个问题。
答案 0 :(得分:0)
您数据框的dtype是“对象”而不是数字。
首先,将数据框中的dtype更改为数字。
df_res.apply(pd.to_numeric).style.background_gradient(cmap=cm)
注意dtypes:
import seaborn as sns
import pandas as pd
cm = sns.light_palette('green', as_cmap=True)
df_res = pd.DataFrame(index =['foo','bar'],columns = ['Value 1','Value 2','Value 3'])
df_res.loc['foo'] = [-.5*100, .3,.2]
df_res.loc['bar'] = [.3*100, .6,.9]
df_res.info()
输出:
<class 'pandas.core.frame.DataFrame'>
Index: 2 entries, foo to bar
Data columns (total 3 columns):
Value 1 2 non-null object
Value 2 2 non-null object
Value 3 2 non-null object
dtypes: object(3)
memory usage: 144.0+ bytes