У меня есть это:
def steps(ns):
return [random.choice([[0, 1], [1, 0], [0, -1], [-1, 0]]) for i in range(ns)]
, которое возвращает, например, для ns = 4: [[-1, 0], [0, -1], [1, 0], [1, 0]]
Я хочу сделать что-то вроде этого:
def Walk(ns):
return sc.cumsum(steps2d(ns))
, чтобы вычислить сумму из вышеприведенного. Я хочу, например, иметь:): [[0, 1], [1, 1], [1, 2], [1, 3], [1, 2], [2, 2], [1, 2], [1, 3], [1, 4], [0, 4]]
С моей функцией Walk я беру что-то вроде:
[1 1 1 2 2 3 3 2 2 3 4 43 3 4 4 4 3 3 4]
Есть ли способ сделать это?Спасибо!