Pandas用列值交换列名

Question

我有一个如下所示的df：

User Chat_People Talk_People Schedule_Time
1    first       second      third
2    second      third       first

我想把它转换成如下形式：

User First         Second       Third
1    Chat_People   Talk_People  Schedule_Time
2    Schedule_Time Chat_People  Talk_People

我试着这样做：

df["First"] = df.apply((lambda x: col_name for col_name in ["Chat_People","Talk_People","Schedule_Time"] if x[col_name] == "first"), axis=1)

但我遇到了：

TypeError: cannot create 'generator' instances

任何帮助都将不胜感激！

Answer 1

先使用melt，然后使用pivot

out = df.melt('User').pivot('User','value','variable').reset_index()
out
value  User          first       second          third
0         1    Chat_People  Talk_People  Schedule_Time
1         2  Schedule_Time  Chat_People    Talk_People