Pandas在dataframe获得df.style后创建一个新列

Question

我试图在dataframe获得df.style后添加一个新列。然而，我收到了一条错误消息：

"样式器"对象不支持项目分配

下面是我的代码：

import pandas as pd
df = pd.DataFrame([[10,3,1], [3,7,2], [2,4,4]], columns=list("ABC"))
subsets = pd.IndexSlice[:, 'A']
df2 = df.style.applymap(lambda x: 'background-color: yellow', subset = subsets)
df2['sum'] = None

以下是我想要实现的目标：

链接

如何将df2转换为可以编辑的真实数据帧？非常感谢。

Answer 1

您不能通过添加数据处理来处理样式器对象，例如添加新列，您需要在以下操作之前添加它：

df2['sum'] = None
subsets = pd.IndexSlice[:, 'A']
df2 = df.style.applymap(lambda x: 'background-color: yellow', subset = subsets)