使用带有多个映射函数的 pandas applymap()

Question

我正在使用df.to_excel() 将数据从 pandas 数据框输出到 Excel。为了提高可读性，我使用df.style.applymap() 根据内容更改单元格的颜色。

假设我有一个看起来像这样的数据框：

df = 
       Account  Revenue   AccountAge
0     "Boeing"     5000          5.6
1   "Lockheed"   -10000          1.2
2     "Airbus"    12000          0.6
3   "Northrop"    -3000          8.4

帐户年龄表示我在账簿上拥有此帐户的时间。也许我想将负收入单元格标记为红色。我能做到：

def color_neg_revenue(val):
    if val < 0:
        color = 'red'
    return "background-color: %s" % color

df = df.style.applymap(color_neg_revenue, subset=["Revenue"])

我可以将它导出到 excel 中，而且格式看起来很棒！但是假设我也想在尝试时将新帐户标记为黄色：

def color_new_account(val):
    if val < 3:
        color = 'yellow'
    return "background-color: %s" % color

df = df.style.applymap(color_new_account, subset=["AccountAge"])

我明白了：

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

为什么我不能在我已经使用过applymap() 的数据帧上使用applymap()？我该如何解决这个问题？

Answer 1

问题不在于函数。在您的第一次通话中，您执行以下操作：

<s><b>df =</b> df.style.applymap(color_neg_revenue, subset=["Revenue"])</s>

所以现在df 不再是DataFrame，而是Styler。这样的Styler 当然没有.style 属性了。因此，您最好将其分配给另一个变量，并在该变量上进行链接。喜欢：

s = df.style.applymap(color_neg_revenue, subset=["Revenue"])

# Look mom, no I have no .style
s = s.applymap(color_new_account, subset=["AccountAge"])

因此，当您使用 Styler 时，您可以链接此类 applymap 调用，但此类样式器本身没有 .style 属性。