python - 在 padas 数据框中写入类似 json 的数据

Question

我有一个数据框：
df = pd.DataFrame({"a":[1,2], "b":[2,1]})
我想创建一个新列，其中包含类似 json 的结构（对象列表）。但问题是，它有不同长度的列表。像这样：
df['z'] = df.apply(lambda row: np.where(row.a>row.b, [{"c":row["b"]}], [{"d":1}, {"e":2}] ), axis=1)
但我得到一个错误

所需的输出如下：

一个	b	z
1	2	[{"d":1}, {"e":2}]
2	1	[{"c":1}]

标签： pythonpandas