파이썬에서 누락 값 imputation

Question

내 데이터 집합에 Imputer.fit_transform()을 적용한 후 변형 된 데이터 프레임에서 열 이름이 손실됩니다. 열 이름을 잃지 않고 그것을 전가 할 방법이 있습니까?

Answer 1

질문에 대한 의견에서 말했듯이, 데이터 프레임의 값을 Imputer에서 반환 된 데이터로 바꾸면됩니다.

말할 수 있습니다이 당신의 dataframe입니다 :

import numpy as np 
import pandas as pd 

df = pd.DataFrame(data=[[1,2,3], 
         [3,4,4], 
         [3,5,np.nan], 
         [6,7,8], 
         [3,np.nan,1]], 
        columns=['A', 'B', 'C']) 

현재 df :

df[df.columns] = Imputer().fit_transform(df) 

경우 :

A B C 
0 1 2.0 3.0 
1 3 4.0 4.0 
2 3 5.0 NaN 
3 6 7.0 8.0 
4 3 NaN 1.0 

당신이 Imputer에 df을 전체 전송하는 경우, 바로 이것을 사용 일부 열만 보내고 그런 다음 해당 열만 사용하여 할당하십시오. N 결과 :

columns_to_impute = ['B', 'C'] 
df[columns_to_impute] = Imputer().fit_transform(df[columns_to_impute]) 

출력 :

 A B C 
0 1.0 2.0 3.0 
1 3.0 4.0 4.0 
2 3.0 5.0 4.0 
3 6.0 7.0 8.0 
4 3.0 4.5 1.0