imr = Imputer(missing_values='NaN', strategy='median', axis=0) imr = imr.fit(data[['age']]) data['age'] = imr.transform(data[['age']]).ravel()