Установить значение по умолчанию для функции

Question

Мне нужно сделать функцию, которая принимает фрейм данных в качестве одного ввода и обязательные столбцы в качестве второго ввода

def get_feature(df,cols=df.columns):
    . . .
    . . . 
    . . .
    return features

поэтому здесь я хочу, чтобы вторым параметром был список столбцов, которые пользователь может ввести, в противном случае он должен принять все столбцы dataframe по умолчанию передаются как один параметр, есть ли способ сделать это?

Answer 1

Вы можете сделать что-то вроде этого:

def get_feature(df, cols=None):
    if cols is None:
        cols = df.columns
    . . . 
    . . .
    return features