generalizing the solution to the 'extract rows with unequal values' exercise

2016-11-29 13:09:51 +01:00
parent 733e2c3204
commit 37e7fe2563
2 changed files with 2 additions and 6 deletions
--- a/exercises.md
+++ b/exercises.md
@@ -1108,9 +1108,7 @@ print(rows)
 # Author: Robert Kern

 Z = np.random.randint(0,5,(10,3))
-E = np.logical_and.reduce(Z[:,1:] == Z[:,:-1], axis=1)
-U = Z[~E]
-print(Z)
+U = Z[Z.max(axis=1) != Z.min(axis=1),:]
 print(U)
 ```