如何在Python的plot.bar中显示交叉表0计数并保持颜色对应？

问题出在默认情况下，pandas交叉表只会保留数据中实际出现的类别，而dropna=False仅处理缺失值(NaN)，对未出现的类别无效。要强制显示全部5个评分类别并保证颜色对应正确，按以下步骤操作：

1. 强制保留所有评分类别 #

将评分列转换为Categorical类型，指定固定的1-5类别集合，这样交叉表会自动补全未出现类别的计数为0：

import pandas as pd
import matplotlib.pyplot as plt

# 假设你的数据存储在df中，评分列名为rating
df['rating'] = pd.Categorical(df['rating'], categories=[1,2,3,4,5], ordered=True)

2. 生成包含全部类别的交叉表 #

基于转换后的分类列生成交叉表，此时即使没有1分的数据，也会显示该类别，计数为0：

# 这里以按分组统计各评分数量为例，可根据你的实际需求调整行/列
cross_tab = pd.crosstab(df['分组列名'], df['rating'], dropna=False)

3. 绑定固定顺序的颜色列表 #

定义评分与颜色的映射关系，然后按1-5的固定顺序生成颜色列表，确保颜色与评分一一对应：

color_mapping = {
    1: 'red',
    2: 'orange',
    3: 'yellow',
    4: 'lightgreen',
    5: 'green'
}
# 按1-5的顺序提取颜色，保证颜色对应正确
bar_colors = [color_mapping[score] for score in [1,2,3,4,5]]

4. 绘制柱状图 #

使用生成的颜色列表绘制柱状图，此时即使某些评分计数为0，也会显示对应的空柱子，且颜色完全匹配：

cross_tab.plot(kind='bar', color=bar_colors)
plt.xlabel('分组')
plt.ylabel('数量')
plt.title('各评分选项数量统计')
plt.legend(title='评分')
plt.show()

关键说明 #

• pd.Categorical是核心：它能强制pandas保留你指定的所有类别，无论数据中是否存在该类别的值，这是dropna=False做不到的（后者仅处理NaN，不处理未出现的非缺失类别）。
• 颜色列表按固定顺序生成：避免因交叉表列顺序变化导致颜色对应错误，确保2分始终对应橙色，不受数据中类别出现顺序影响。

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