如何在Pandas中将MultiIndex列转为字符串列表而非元组列表?
解决Pandas MultiIndex列名直接转字符串列表的问题
当然有更高效的办法,完全不用事后手动编辑列表!针对MultiIndex类型的列名,这里有几种直接生成纯字符串元素列表的方案,适配不同场景:
通用方案(适配单层/多层MultiIndex):用列表推导式结合
str.join(),把每个列名元组的元素拼接成单个字符串。如果是多层列名,还可以自定义分隔符(比如下划线、点号):# 用下划线拼接多层列名,单层的话自动返回单个字符串 columns = ['_'.join(col_tuple) for col_tuple in df.columns]举个例子,如果你的列名是
MultiIndex([('A',), ('B', 'Sub1'), ('C', 'Sub2')]),执行后会得到['A', 'B_Sub1', 'C_Sub2']。更简洁的映射写法:用
df.columns.map()配合lambda函数,效果和上面一致,但代码更紧凑:columns = df.columns.map(lambda x: '_'.join(x))单层MultiIndex专属方案:如果你确定列名只有一层(也就是每个元组只有一个元素,比如
('A',)),可以直接提取对应层级的字符串:columns = df.columns.get_level_values(0).tolist()这个方法跳过了元组处理,直接拿到纯字符串列表,效率更高。
这些方法都能一步到位生成你需要的字符串列表,不用再做后续的列表编辑工作~
内容的提问来源于stack exchange,提问作者matthewr
相关产品推荐
相关产品推荐

