Consecutive if-conditions in function, Python

Question

I am creating a function which calulates an outcome depending on parameters in the formula.

Separately to the first if-function I'd like to add a second condition for cases in which n is not None

def target_n(data, y, q = None, s = 1, n = None, ascending = True):
        if q is not None:
            target = quantile(data, q, s)
            return(target)
        else:
            target = y(data,s)
            return(target)
        if n is not None:
            sort = target.sort(tst.columns[0],ascending = ascending).ix[0:n,:]
        return(sort)

Nevertheless the function returns "target" and not "sort" for cases in which n is not None. How can I implement this?

Julien Marrec · Accepted Answer · 2016-11-07 12:26:55Z

1

Your problem is that you have a return statement whether q is None or not. As soon as the execution reaches the return statement, it exits the function.

You could do something like this

def target_n(data, y, q = None, s = 1, n = None, ascending = True):
    if q is not None:
        target = quantile(data, q, s)
    else:
        target = y(data,s)
    if n is not None:
        target = target.sort(tst.columns[0],ascending = ascending).ix[0:n,:]

    return(target)

answered Nov 7, 2016 at 12:26

Julien Marrec

11.9k5 gold badges51 silver badges67 bronze badges

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

Collectives™ on Stack Overflow

Consecutive if-conditions in function, Python

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