Заполняем пропуски - варианты

######################################################################################################
data = pd.read_csv('data.csv')

data_grouped = data.groupby('income_type')['total_income'].mean()

def func(row):
    if pd.isna(row['total_income']):
        return data_grouped.loc[row['income_type']]
    return row['total_income']

data['total_income'] = data.apply(func, axis=1)

######################################################################################################

data = pd.read_csv('data.csv')

data_transformed = data.groupby('income_type')['total_income'].transform('mean')

data['total_income'] = data['total_income'].fillna(data_transformed)

######################################################################################################

data = pd.read_csv('data.csv')

data['total_income'] = data.groupby('income_type')['total_income'].apply(lambda s: s.fillna(s.mean()))
######################################################################################################