我試圖使用 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)
uj5u.com熱心網友回復:
您可以先使用 來創建差異陣列np.random.choice,然后使用 來累計每個專案np.cumsum():
np.cumsum(np.random.choice([1, -1], size=10))
轉載請註明出處,本文鏈接:https://www.uj5u.com/qita/336662.html
上一篇:將物件陣列的結構更改為所需格式
