Python - 数据框合并 + JSON

Question

| |id|a|b
|1| 5|1|4
|2|10|2|5
|3|15|3|6
     +
| |id|a |b
|1| 5|10|13
|2|10|11|14
|3|15|12|15
     =
| |id|  a  |  b
|1| 5|1, 10|4, 13
|2|10|2, 11|5, 14
|3|15|3, 12|6, 15

我正在尝试通过 id 合并两个数据帧并将结果保存在 json 文件中。

Answer 1

使用pd.concat和df.groupby和df.agg

df3 = pd.concat([df1, df2]).groupby('id', as_index=False).agg({'a': lambda x: ', '.join(map(str, x)), 'b': lambda x: ', '.join(map(str, x))})
print(df3)

输出：

   id      a      b
0   5  1, 10  4, 13
1  10  2, 11  5, 14
2  15  3, 12  6, 15

您可以使用df.to_json()转换为 json

print(df3.to_json())

输出：

{
  "id": {
    "0": 5,
    "1": 10,
    "2": 15
  },
  "a": {
    "0": "1, 10",
    "1": "2, 11",
    "2": "3, 12"
  },
  "b": {
    "0": "4, 13",
    "1": "5, 14",
    "2": "6, 15"
  }
}