pandas - 删除 Pandas pd.to_dict 中的空白键条目

Question

Pandas 有一个非常好的功能，可以通过 pd.to_dict('records') 将我们的数据帧导出到字典列表中。 .

例如:

d = pd.DataFrame({'a':[1,2,3], 'b':['a', 'b', None]}) 
d.to_dict('records')

返回

[{'a': 1, 'b': 'a'}, {'a': 2, 'b': 'b'}, {'a': 3, 'b': None}]

对于我的用例，我更喜欢以下条目:

[{'a': 1, 'b': 'a'}, {'a': 2, 'b': 'b'}, {'a': 3}]

在哪里可以看到 b的 key 从第三个条目中删除。这是使用 jsonlite 时 R 中的默认行为，并且想知道如何从每个条目中删除缺少值的键。

Answer 1

我们可以做 stack

l=d.stack().groupby(level=0).agg(lambda x : x.reset_index(level=0,drop=True).to_dict()).tolist()
Out[142]: [{'a': 1, 'b': 'a'}, {'a': 2, 'b': 'b'}, {'a': 3}]