使用 Styler (pandas) 为数据框着色后格式化数字

Question

我在 pandas 中创建了一个 DataFrame，我想使用颜色索引为单元格着色（低值红色，高值绿色）。我成功地这样做了，但是着色使我无法格式化单元格。

import pandas as pd

df = pd.DataFrame({'a': [0.5,1.5, 5],
                   'b': [2, 3.5, 7] })

df = df.style.background_gradient(cmap='RdYlGn')
df

返回

但是，当我尝试使用df.round(2) 来格式化数字时，会弹出以下错误：

AttributeError: 'Styler' object has no attribute 'round'

有没有人可以告诉我出了什么问题？

Answer 1

看看pandas styling guide。 df.style 属性返回 Styler 实例，而不是数据框。从 pandas 样式指南中的示例来看，似乎数据框操作（如舍入）是先完成的，而样式是最后完成的。 pandas 样式指南中有a section on precision。该部分提出了三种不同的选项来显示值的精度。

import pandas as pd

df = pd.DataFrame({'a': [0.5,1.5, 5],
                   'b': [2, 3.5, 7] })

# Option 1. Round before using style.
style = df.round(2).style.background_gradient(cmap='RdYlGn')
style

# Option 2. Use option_context to set precision.
with pd.option_context('display.precision', 2):
    style = df.style.background_gradient(cmap='RdYlGn')
style

# Option 3. Use .set_precision() method of styler.
style = df.style.background_gradient(cmap='RdYlGn').set_precision(2)
style