擅长:python、mysql、java
<p>这是另一个答案。</p>
<pre><code>N = 3 # window size
m = np.array([
[[0, 10],
[0, 0]],
[[0, 0],
[1, 1.]],
[[0, 0],
[0, 1.]],
[[0, 5],
[0, 0]],
[[0, 10],
[0, 0]]])
print m.shape
for t in range(m.shape[0]-N):
print '\nwindow:', t, 'to', t+N-1
print m[t:t+N,:,:].shape
print m[t:t+N,:,:].sum(axis=0)/(N)
</code></pre>
<p>输出为</p>
<pre><code>(5, 2, 2)
window 0 to 2
(3, 2, 2)
[[ 0. 3.33333333]
[ 0.33333333 0.66666667]]
window: 1 to 3
(3, 2, 2)
[[ 0. 1.66666667]
[ 0.33333333 0.66666667]]
</code></pre>