Pandas pivot_table后使用Styler.applymap触发KeyError问题求助

问题原因 #

你碰到的KeyError是因为pivot_table生成的DataFrame存在非唯一的索引或列名，而Pandas的Styler组件的apply/applymap方法不支持在这类结构上运行。

解决步骤 #

• 先排查重复项
  运行以下代码确认是索引还是列重复：

  # 检查索引是否唯一
  print(df.index.is_unique)
  # 检查列是否唯一
  print(df.columns.is_unique)
  

• 修复非唯一结构
  针对不同情况处理：

  1. 索引重复：
     最简单的方式是重置索引，把原索引转为普通列：

     df = df.reset_index()
     

     如果要保留索引但区分重复项，可添加序号后缀：

     df.index = [f"{idx}_{i}" for i, idx in enumerate(df.index)]
     

  2. 列重复：
     给重复列名添加序号后缀：

     col_counts = df.columns.value_counts()
     new_cols = []
     for col in df.columns:
         if col_counts[col] > 1:
             # 记录当前列出现的次数
             count = new_cols.count(col)
             new_cols.append(f"{col}_{count+1}")
         else:
             new_cols.append(col)
     df.columns = new_cols
     

  3. 从根源避免：
     生成pivot_table时，确保index、columns参数的组合不会产生重复项，或者调整aggfunc（比如aggfunc='sum'）来聚合重复的数值，避免行列重复。

• 重新应用样式
  确认索引和列都唯一后，再执行你的样式代码即可正常运行。

内容的提问来源于stack exchange，提问作者Nichi ek