python - Pandas:组合多索引数据帧的标题行

Question

假设我有一个 DataFrame 如下:

first   bar       baz       foo
second  one  two  one  two  one  two three
A       1    2    3    4    5    6   7
B       8    9    10   11   12   13  14

我想像这样创建一个新的 DataFrame:

        barone  bartwo  bazone baztwo  fooone footwo foothree
A       1       2       3      4       5      6      7
B       8       9       10     11      12     13     14

可能的代码是什么？

Answer 1

1.使用Python 3.6+更新使用f-string formatting使用列表理解:

df.columns = [f'{i}{j}' for i, j in df.columns]

2.使用 map和 join :

df.columns = df.columns.map(''.join)

3. 如果您的列具有数字数据类型，请使用 map和 format :

df.columns = df.columns.map('{0[0]}{0[1]}'.format) 

输出:

   barone  bartwo  bazone  baztwo  fooone  footwo  foothree
A       1       2       3       4       5       6         7
B       8       9      10      11      12      13        14