print(df.isnull().sum())

df_sotr_avg = df[df['income_type'] == 'сотрудник']['total_income'].mean()
df_comp_avg = df[df['income_type'] == 'компаньон']['total_income'].mean()
df_pens_avg = df[df['income_type'] == 'пенсионер']['total_income'].mean()
df_gos_avg = df[df['income_type'] == 'госслужащий']['total_income'].mean()

if (df['income_type'] == 'сотрудник'):
    df['total_income'] = df['total_income'].fillna(df_sotr_avg)
elif (df['income_type'] == 'компаньон'):
    df['total_income'] = df['total_income'].fillna(df_comp_avg)
elif (df['income_type'] == 'пенсионер'):
    df['total_income'] = df['total_income'].fillna(df_pens_avg)
elif (df['income_type'] == 'госслужащий'):
    df['total_income'] = df['total_income'].fillna(df_gos_avg)

# ------------------------------------------------------------------------

df_sotr_avg = df[df['income_type'] == 'сотрудник']['days_employed'].mean()
df_comp_avg = df[df['income_type'] == 'компаньон']['days_employed'].mean()
df_pens_avg = df[df['income_type'] == 'пенсионер']['days_employed'].mean()
df_gos_avg = df[df['income_type'] == 'госслужащий']['days_employed'].mean()

if (df['income_type'] == 'сотрудник'):
    df['days_employed'] = df['days_employed'].fillna(df_sotr_avg)
elif (df['income_type'] == 'компаньон'):
    df['days_employed'] = df['days_employed'].fillna(df_comp_avg)
elif (df['income_type'] == 'пенсионер'):
    df['days_employed'] = df['days_employed'].fillna(df_pens_avg)
elif (df['income_type'] == 'госслужащий'):
    df['days_employed'] = df['days_employed'].fillna(df_gos_avg)