Pandas: Как вернуть один столбец из двух переданных в DataFrame.apply()?

Максим Припадчев, Вы правы, ваш код работает. Но в моей таблице есть еще столбцы, я их убрал, т.к. смысловой нагрузки для вопроса они не имеют.
Так вот, стоит добавить еще 1 столбец и в Ваш тестовый DF, то в такой конструкции вылетает ошибка:
ValueError: Columns must be same length as key

Pandas: Как вернуть один столбец из двух переданных в DataFrame.apply()?

Спасибо большое, идею я понял, недавно только начал изучать pandas.
Единственное код выдавал ошибку, так как пытался записать несколько столбцов в 1.
Сработало так:

data['Возраст'] = data.groupby('Социальный класс')['Возраст'].transform(lambda x: x.fillna(x.mean()))