Python Pandas: переименование столбцов после поворота

Question

У меня есть кадр данных панд, который поворачивается.Я не знаю, как изменить имена столбцов, чтобы я мог продолжить работу с поворотным фреймом данных естественным образом.Ниже приведен минимальный рабочий пример.

df = pd.DataFrame({'foo': ['one', 'one', 'one', 'two', 'two',
                            'two'],
                    'bar': ['A', 'B', 'C', 'A', 'B', 'C'],
                    'baz': [1, 2, 3, 4, 5, 6],
                    'zoo': ['x', 'y', 'z', 'q', 'w', 't']})

После поворота с

df.pivot(index='foo', columns='bar', values=['baz', 'zoo'])

вывод:

      baz       zoo
bar   A  B  C   A  B  C
foo
one   1  2  3   x  y  z
two   4  5  6   q  w  t

Что будет следующим шагом, чтобы получить вывод ниже?

   A_baz  B_baz  C_baz   A_zoo  B_zoo  C_zoo

one   1    2       3       x      y      z
two   4    5       6       q      w      t

Большое спасибо!

Answer 1

Используйте f-string s с пониманием списка:

#python 3.6+
df.columns = [f'{j}_{i}' for i, j in df.columns]
#lower python versions
#df.columns = ['{}_{}'.format(j, i) for i, j in df.columns]
print (df)
    A_baz B_baz C_baz A_zoo B_zoo C_zoo
foo                                    
one     1     2     3     x     y     z
two     4     5     6     q     w     t

или DataFrame.swaplevel с map и join:

df = df.pivot(index='foo', columns='bar', values=['baz', 'zoo']).swaplevel(0,1,axis=1)

df.columns = df.columns.map('_'.join)
print (df)
    A_baz B_baz C_baz A_zoo B_zoo C_zoo
foo                                    
one     1     2     3     x     y     z
two     4     5     6     q     w     t