Вставка переменных в DataFrame pandas

Есть строка в обработке данных датафрейма (чистка пустых ячеек)

records_clear = (records.loc[records['Style'].notna(), ['Age', 'Style']])

Как вместо Style и Age подставить переменные? Пытаюсь сделать через f-string.

records_clear = f"(records.loc[records['{style_var}'].notna(), ['{age_var}', '{style_var}']])"

Но в таком случае переменная records_clear меняет свой тип на str. Помогите.


Ответы (1 шт):

Автор решения: SergFSM

просто подставляйте:

age_var = 'Age'
records_clear = (records.loc[records['Style'].notna(), [age_var, 'Style']])
→ Ссылка