使用for循环批量生成分类变量countplot如何仅筛选category类型列

错误原因 #

• 直接将列dtype与字符串'category'比对的兼容性差，不同pandas版本、有序分类场景下可能无法正确识别分类列，导致筛选逻辑失效
• 绘图代码行末尾多余的逗号会将返回值封装为元组，干扰matplotlib的画布状态管理，最终只能渲染出第一张图

修正方案 #

你可以直接用pandas内置的select_dtypes方法筛选分类列，不需要手动判断类型，代码更简洁也更稳定，修正后代码如下：

# 直接筛选所有category类型的列循环绘图
for col in penguins.select_dtypes('category').columns:
    plt.figure(figsize=(16,8))
    sns.countplot(x=col, data=penguins, palette='flare')
    plt.show()

如果需要保留原有的判断逻辑写法，将类型判断替换为pandas官方提供的类型校验方法即可：

import pandas as pd

for i in penguins.columns:
    # 用官方API判断是否为分类类型，兼容性更好
    if pd.api.types.is_categorical_dtype(penguins[i]):
        plt.figure(figsize=(16,8))
        sns.countplot(x=i, data=penguins, palette='flare')
        plt.show()

额外验证 #

如果运行后还是没有输出对应图表，可以先执行以下代码确认数据集是否真的存在category类型的列：

print(penguins.dtypes)

如果目标分类列显示为object类型，需要先通过penguins[col] = penguins[col].astype('category')转换为分类类型后再执行绘图逻辑。

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