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

为了从上面计算 cumsum。我想要例如：

Walk(10) : [[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 4 3 3 4 4 4 3 3 4]

有办法吗？ 谢谢！

Answer 1

试试scipy.cumsum(..., axis=0)。