Merge pull request #148 from Sword-holder/master

Fix solution of question 76.
rougier · Apr 26, 2021 · 22f47a6 · 22f47a6
2 parents 3135668 + 86e24f1
commit 22f47a6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)