SciPy的 - 從對列表計算cumsum

Question

我有這樣的：

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)) 

爲了從above.I計算cumsum想以例如有：

步行（10 ）：[[0,1] ，[1,1]，[1,2]，[1,3]，[1,2]，[2,2]，[1,2]，[1,3]，[1, 4]， [0,4]

隨着礦井我正在某事物等的步行功能：

[1 1 1 2 2 3 3 2 2 3 4 4 3 3 4 4 4 3 3 4]

有沒有辦法做到這一點？ 謝謝！

Answer 1

嘗試scipy.cumsum(..., axis=0)。