相关疑难解决方法(0)

似乎没有简单计算numpy/scipy上的移动平均值的函数,导致复杂的解决方案.

我的问题是双重的:

• 用numpy(正确)实现移动平均线的最简单方法是什么？
• 由于这看起来非常重要且容易出错,因此有充分的理由不在这种情况下包含电池吗？

我有一个矩阵(准确地说是2d numpy ndarray):

A = np.array([[4, 0, 0],
              [1, 2, 3],
              [0, 0, 5]])

我想A根据另一个数组中的滚动值独立滚动每一行:

r = np.array([2, 0, -1])

也就是说,我想这样做:

print np.array([np.roll(row, x) for row,x in zip(A, r)])

[[0 0 4]
 [1 2 3]
 [0 5 0]]

有没有办法有效地做到这一点？也许使用花哨的索引技巧？