根据对 Dataframe 的查找将列替换为多级列

Question

如何替换这个单列标题：

    foo bar
0   0   0
1   0   0

要获取这些多级列：

基于此数据帧映射

    col1    col2    col3
0   foo     a       A
1   bar     b       B
2   baz     c       C

我正在尝试使用列表理解来创建一个新的多级列索引，但似乎没有用...我感觉还有一种更 pythonic 的方法可以实现这一点

df1 = pd.DataFrame({'foo':[0,0],
                    'bar':[0,0]})
df2 = pd.DataFrame({'col1':['foo','bar','baz'],
                    'col2':['A','B','C'],
                    'col3':['a','b','c']})

df1.columns = [(df2.loc[df2['col1']==i,'col2'], df2.loc[df2['col1']==i,'col3']) for i in df1.columns]

Answer 1

您可以将 df2 转换为一系列元组并将其映射到列：

df1.columns = df1.columns.map(df2.set_index('col1').apply(tuple, axis=1))

输出：

根据对 Dataframe 的查找将列替换为多级列

Replace columns with multi level columns based on lookup on Dataframe

python

hierarchical-data

pandas