Set default value of a function

Question

I have to make a function which takes dataframe as one input and required columns as second input

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

so here i want second parameter to be a list of columns which user can enter otherwise it should take all the columns of dataframe passed as parameter one as default, is there any way to do that?

Heike · Accepted Answer · 2020-03-15 04:50:12Z

2

You could do something like this:

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

answered Mar 15, 2020 at 4:50

Heike

24.5k2 gold badges34 silver badges46 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Set default value of a function

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related