с пропусками. Есть фича Sex со значением 1 или 2. Хочу заполнить пропуски медианными значениями, но чтобы они соответствовали фиче Sex. То есть если заполняю NaN в строчках, где Sex = 1, то хочу заполнить медианой для этой фичи при Sex = 1. Фич много, пошёл по циклу.
for col in data[data['Sex'] == 1.0].columns[2:]:
data[data['Sex'] == 1.0][col].fillna((data[data['Sex'] == 1.0][col]).median(), inplace = True)
Но я явно делаю что-то не так. Как переделать код?
ты хочешь категориальную фитчу заполнить средним значением? Предположим у тебя 4 наблюдения: 1 2 1 Nan - ты хочешь вместо Nan поставить 4/3?
Обсуждают сегодня