Return multiple values with function using Python

Question

I have a list of code to update the value of time for each 5 elements. This is the code:

msg = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15]
theta=1
time=4.5

def get_new_theta(msg, theta,time):
    new_theta = [theta]
    new_time = [time]
    for a, b in zip(msg[3::5], msg[4::5]):
        new_theta.append(new_theta[-1] + a + b)
        new_time.append(new_time[-1]+time)
    return new_theta
    return new_time

for i, theta in enumerate(get_new_theta(msg, theta,time)[:-1]):
    for j in range(5):
        print(theta, msg[i*5+j],time)

However, I did not get the desired output. The desired output should be:

Need help from this forum. Thank you.

You have two return statements but only one will ever be reached. — John Forbes
– John Forbes, Commented Aug 21, 2018 at 6:41
@ParthS007. I have try it but it is not same as the desired output. — Afir
– Afir, Commented Aug 21, 2018 at 6:52

xdze2 · Accepted Answer · 2018-08-21 08:53:49Z

1

def get_new_theta(msg, theta, time):
    new_theta = [theta]
    new_time = [time]
    for a, b in zip(msg[3::5], msg[4::5]):
        new_theta.append(new_theta[-1] + a + b)
        new_time.append(new_time[-1]+time)
    return new_theta[:-1], new_time[:-1]

msg = [1,2,3,4,5,6,7,8,9,10,11,12,13,14,15]
theta = 1
time = 4.5

for i, (theta, time) in enumerate( zip(*get_new_theta(msg, theta, time)) ):
    for j in range(5):
        print(theta, msg[i*5+j], time)

gives:

get_new_theta(msg, theta, time) returns a tuple of the lists (new_theta, new_time). And zip(*get_new_theta(msg, theta, time)) gives an iterator over tuples (theta, time)

answered Aug 21, 2018 at 8:53

xdze2

4,1612 gold badges14 silver badges30 bronze badges

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

Collectives™ on Stack Overflow

Return multiple values with function using 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