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