將具有多行的Pandas數據框轉換為一行

Question

我需要將具有多行的Pandas數據框轉換為一行。

具有以下數據框。

X     Y    V
A     0    0.1
A     1    0.2
B     0    0.3
B     1    0.4

我想將數據框轉換為以下格式。 V_0表示“ Y=0值”。 如何實現這種轉型？

X     V_0    V_1
A     0.1    0.2
B     0.3    0.4

Answer 1

使用pd.DataFrame.pivot ：

res = df.pivot(index='X', columns='Y').add_prefix('V_')
res.columns = res.columns.droplevel(0)
res = res.reset_index()

print(res)

Y  X  V_0  V_1
0  A  0.1  0.2
1  B  0.3  0.4