python - 如何按列对 pandas 中的数据框进行分组并获取 Json 数组中的所有其他列数据？

Question

我有一个 pandas 数据框:

df = pd.DataFrame({
        "Column1": ['A', 'A', 'B', 'B', 'C', 'C', 'C'],
        "Column2": ['x1', 'x2', 'z1', 'z2', 'm1', 'm2', 'm3'],
        "Column3": ['y1', 'y2', 'd1', 'd2', 'n1', 'n2', 'n3']
})

如何按第一列分组以获得以下输出

第 1 列 |新列 -------- | --------|-------- 一个 | [{“列2”:“x1”，“列3”:“y1”}，{“列2”:“x2”，“列3”:“y2”}] 乙| [{“列2”:“z1”，“列3”:“d1”}，{“列2”:“z2”，“列3”:“d2”}] C | [{“列2”:“m1”，“列3”:“n1”}，{“列2”:“m2”，“列3”: "m2"},{"Column2": "m3", "Column3": "n3"}]

我需要这样做才能将另一个数据框与 Column1 连接。

谢谢

Answer 1

您可以使用以下代码:

df_out =  df.groupby('Column1')\
            .apply(lambda x: x.drop('Column1',axis=1).to_dict(orient='record'))\
            .reset_index(name='new_column')

输出:

  Column1                                         new_column
0       A  [{'Column3': 'y1', 'Column2': 'x1'}, {'Column3...
1       B  [{'Column3': 'd1', 'Column2': 'z1'}, {'Column3...
2       C  [{'Column3': 'n1', 'Column2': 'm1'}, {'Column3...

使用 to_dict 显示完整数据框

df_out.to_dict()

输出:

 'A': [{'Column2': 'x1', 'Column3': 'y1'}, {'Column2': 'x2', 'Column3': 'y2'}],
 'B': [{'Column2': 'z1', 'Column3': 'd1'}, {'Column2': 'z2', 'Column3': 'd2'}],
 'C': [{'Column2': 'm1', 'Column3': 'n1'},
  {'Column2': 'm2', 'Column3': 'n2'},
  {'Column2': 'm3', 'Column3': 'n3'}]}