Create numpy array with conditional step

Question

Im trying to convert this code into one-liner function using lambda and numpy, the function should return numpy array, but i dont know how to save the step value for each iteration

import random

def generate_array(n: int):
    step = 1
    arr = []
    
    for i in range(n):
        step = step + 1 if random.choice([True, False]) else step - 1 
        arr .append(step)

    return arr

generate_array(10)

QuantumMecha · Accepted Answer · 2021-10-26 06:34:24Z

1

You can first create the difference array using np.random.choice, and then sum each item up cumulatively using np.cumsum():

np.cumsum(np.random.choice([1, -1], size=10))

edited Oct 26, 2021 at 6:34

answered Oct 26, 2021 at 6:32

QuantumMecha

1,5417 silver badges22 bronze badges

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

1 Comment

Nils Werner Over a year ago

There is no need for a list comprehension here

Collectives™ on Stack Overflow

Create numpy array with conditional step

1 Answer 1

1 Comment

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

1 Comment

Your Answer

Sign up or log in

Post as a guest

Related