Fix solution of question 76.

2021-04-21 16:11:23 +08:00 · 2021-04-21 16:11:23 +08:00 · 86e24f1da9
commit 86e24f1da9
parent 3135668de3
1 changed files with 1 additions and 1 deletions
--- a/source/exercises100.ktx
+++ b/source/exercises100.ktx
@ -1041,7 +1041,7 @@ from numpy.lib import stride_tricks

 def rolling(a, window):
    shape = (a.size - window + 1, window)
-    strides = (a.itemsize, a.itemsize)
+    strides = (a.strides[0], a.strides[0])
    return stride_tricks.as_strided(a, shape=shape, strides=strides)
 Z = rolling(np.arange(10), 3)
 print(Z)