У меня есть некоторый pandas DataFrame с NaN в нем. Вот так:
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
Теперь я хочу изгнать из него и в то же время удалить NaNs. Результат должен выглядеть следующим образом:
{'A': {1: 2, 2: 3, 3: 4}, 'B': {2: 44.0}}
Но использование функции pandas to_dict дает мне такой результат:
>>> data.to_dict()
{'A': {1: 2, 2: 3, 3: 4}, 'B': {1: nan, 2: 44.0, 3: nan}}
Итак, как сделать dict из DataFrame и избавиться от NaNs?