我有一些带有NaNs的pandas DataFrame.像这样:
import pandas as pd
import numpy as np
raw_data={'A':{1:2,2:3,3:4},'B':{1:np.nan,2:44,3:np.nan}}
data=pd.DataFrame(raw_data)
>>> data
A B
1 2 NaN
2 3 44
3 4 NaN
Run Code Online (Sandbox Code Playgroud)
现在我想用它制作一个字典,同时删除NaN.结果应如下所示:
{'A': {1: 2, 2: 3, 3: 4}, 'B': {2: 44.0}}
Run Code Online (Sandbox Code Playgroud)
但是使用pandas to_dict函数给我一个这样的结果:
>>> data.to_dict()
{'A': {1: 2, 2: 3, 3: 4}, 'B': {1: nan, 2: 44.0, 3: nan}}
Run Code Online (Sandbox Code Playgroud)
那么如何从DataFrame中制作一个字典并摆脱NaN?