Seaborn Jointplot处理多级索引列时触发TypeError的技术问询

嘿，我来帮你搞定这个报错！问题根源在于你分组聚合后得到的是多级列索引的DataFrame，而Seaborn的jointplot没法直接识别['col2','nunique']这种格式的列引用——它会把这个列表当成普通数组，而不是DataFrame里的列名，所以才触发了TypeError。

咱先看看聚合后的DataFrame列结构：

print(df.columns)
# 输出是：MultiIndex([('col2', 'nunique'), ('col2', 'count')], )

下面给你两种可行的解决方案：

方案1：扁平化多级列索引 #

把两级列名合并成单层级的，比如用下划线连接，这样就能像普通列一样引用了：

import pandas as pd
import seaborn as sns

# 原始数据
df = pd.DataFrame({'col1': ['A', 'A', 'B', 'B', 'B'], 'col2': ['A1', 'B1', 'B1', 'B1', 'A1']})
# 分组聚合
df = df.groupby(['col1']).agg({'col2': ['nunique','count']})
# 扁平化列名
df.columns = ['_'.join(col) for col in df.columns]
# 现在列名是col2_nunique和col2_count，直接传字符串就行
sns.jointplot(x='col2_nunique', y='col2_count', data=df, kind='scatter')

方案2：直接传递元组指定多级列（需Seaborn版本≥0.11.0） #

如果你的Seaborn版本够新（0.11.0及以上），可以直接用元组来指定多级列的位置，这样不用修改原DataFrame的列结构：

import pandas as pd
import seaborn as sns

# 原始数据和聚合步骤不变
df = pd.DataFrame({'col1': ['A', 'A', 'B', 'B', 'B'], 'col2': ['A1', 'B1', 'B1', 'B1', 'A1']})
df = df.groupby(['col1']).agg({'col2': ['nunique','count']})
# 用元组指定多级列
sns.jointplot(x=('col2', 'nunique'), y=('col2', 'count'), data=df, kind='scatter')

两种方法都能解决这个错误，选哪个看你需求就好～

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