使用条件步骤创建 numpy 数组

Question

我试图使用 lambda 和 numpy 将此代码转换为单行函数，该函数应返回 numpy 数组，但我不知道如何保存每次迭代的步长值

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)

Answer 1

您可以首先使用np.random.choice创建差异数组，然后使用np.cumsum()每个项目进行累积np.cumsum() ：

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