重塑Pandas DataFrame：从一列创建多个列

Question

我想重塑图形数据框

​

​//i.sstatic.net/kzqla.png“ alt =“在此处输入图像描述”>

有人可以帮我吗？

Answer 1

您是否尝试过df.pivot（）或pd.pivot（）？列C中的值将成为列标题。之后，将多指数列缩放并重命名它们。

import pandas as pd

#df = df.pivot(['A', 'B'], columns='C').reset_index()    #this also works
df = pd.pivot(data=df, index=['A', 'B'], columns='C').reset_index()
df.columns = ['A', 'B', 'X', 'Y']
print(df)

有时输出

   A   B  X  Y
0  a  aa  1  5
1  b  bb  6  2
2  c  cc  3  7
3  d  dd  8  4

可能会有具有相同索引的重复记录，然后您必须改用pd.pivot_table（）。 param aggfunc = np.mean将占据这些重复记录的平均值，并成为从输出中看到的float。

import pandas as pd
import numpy as np

df = pd.pivot_table(data=df, index=['A', 'B'], columns='C', aggfunc=np.mean).reset_index()
df.columns = ['A', 'B', 'X', 'Y']
print(df)

输出

   A   B    X    Y
0  a  aa  1.0  5.0
1  b  bb  6.0  2.0
2  c  cc  3.0  7.0
3  d  dd  8.0  4.0

Answer 2

您可以尝试

out = df.pivot(index=['A', 'B'], columns='C', values='D').reset_index()

print(out)

C  A   B  X  Y
0  a  aa  1  5
1  b  bb  6  2
2  c  cc  3  7
3  d  dd  8  4