diff --git a/100 Numpy exercises.ipynb b/100 Numpy exercises.ipynb index fdb111b..dbfeb20 100644 --- a/100 Numpy exercises.ipynb +++ b/100 Numpy exercises.ipynb @@ -2157,9 +2157,13 @@ "# Author: Robert Kern\n", "\n", "Z = np.random.randint(0,5,(10,3))\n", - "E = np.logical_and.reduce(Z[:,1:] == Z[:,:-1], axis=1)\n", - "U = Z[~E]\n", "print(Z)\n", + "# solution for arrays of all dtypes (including string arrays and record arrays)\n", + "E = np.all(Z[:,1:] == Z[:,:-1], axis=1)\n", + "U = Z[~E]\n", + "print(U)\n", + "# soluiton for numerical arrays only, will work for any number of columns in Z\n", + "U = Z[Z.max(axis=1) != Z.min(axis=1),:]\n", "print(U)" ] }, diff --git a/100 Numpy exercises.md b/100 Numpy exercises.md index 6ba7458..3bbc780 100644 --- a/100 Numpy exercises.md +++ b/100 Numpy exercises.md @@ -1119,9 +1119,13 @@ 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) +# solution for arrays of all dtypes (including string arrays and record arrays) +E = np.all(Z[:,1:] == Z[:,:-1], axis=1) +U = Z[~E] +print(U) +# soluiton for numerical arrays only, will work for any number of columns in Z +U = Z[Z.max(axis=1) != Z.min(axis=1),:] print(U) ```