按 python 中的 dataframe 分組，並在多列上連接字符串

Question

我有 dataframe 如下所示

A,B,C,D
91102,1,john,
91102,2,john,
91102,3,john,
91102,1,,mary
91102,2,,mary
91102,3,,mary
91103,1,sarah,
91103,2,sarah,
91103,3,sarah,
91103,1,,khan
91103,2,,khan
91103,3,,khan

我想要 groupby A 列和 B 列，並希望得到所需的 output 如下所示

A,B,C,D
91102,1,john,mary
91102,2,john,mary
91102,3,john,mary
91103,1,sarah,khan
91103,2,sarah,khan
91103,3,sarah,khan

我在下面嘗試但沒有給出想要的 output

df=df.groupby(['A', 'B'], as_index=False).agg('' .join)

Answer 1

在groupby中，您可以回填然后取該組的第一行。

df.groupby(['A','B'], as_index=False).apply(lambda x: x.bfill().iloc[0])

結果

       A  B      C     D
0  91102  1   john  mary
1  91102  2   john  mary
2  91102  3   john  mary
3  91103  1  sarah  khan
4  91103  2  sarah  khan
5  91103  3  sarah  khan

Answer 2

嘗試：

x = df.set_index(["A", "B"]).stack().unstack().reset_index()
print(x)

印刷：

       A  B      C     D
0  91102  1   john  mary
1  91102  2   john  mary
2  91102  3   john  mary
3  91103  1  sarah  khan
4  91103  2  sarah  khan
5  91103  3  sarah  khan