python - 扩 Pandas 数据框

Question

我有这样的数据:

Box,Code
Green,1221
Green,8391
Red,3709
Red,2911
Blue,9820
Blue,4530

使用 pandas 数据框，我想知道是否可以输出如下内容:

Box,Code1,Code2
Green,1221,8391
Red,3709,2911
Blue,9820,4530

我的数据每个“框”的行数始终相同。

我一直在 pandas 中试验数据透视表和交叉表(以及堆叠和取消堆叠)，但没有找到任何能让我得到我正在寻找的“扩展”结果的东西。

Answer 1

您可以使用 groupby对于 list 和 DataFrame 构造函数:

a = df.groupby('Box')['Code'].apply(list)
df = pd.DataFrame(a.values.tolist(), index=a.index).add_prefix('Code').reset_index()
print (df)
     Box  Code0  Code1
0   Blue   9820   4530
1  Green   1221   8391
2    Red   3709   2911

或cumcount对于带有 pandas.pivot 的新 Series :

g = df.groupby('Box').cumcount()
df = pd.pivot(index=df['Box'], columns=g, values=df['Code']).add_prefix('Code').reset_index()
print (df)
     Box  Code0  Code1
0   Blue   9820   4530
1  Green   1221   8391
2    Red   3709   2911

与 unstack 类似的解决方案:

df['g'] = df.groupby('Box').cumcount()
df = df.set_index(['Box', 'g'])['Code'].unstack().add_prefix('Code').reset_index()
print (df)
g    Box  Code0  Code1
0   Blue   9820   4530
1  Green   1221   8391
2    Red   3709   2911