Merge pull request #34 from ibah/master

generalizing the solution to the 'extract rows with unequal values'
This commit is contained in:
Nicolas P. Rougier 2016-12-01 13:57:58 +01:00 committed by GitHub
commit f4d6512cef
2 changed files with 12 additions and 4 deletions

View File

@ -2157,9 +2157,13 @@
"# Author: Robert Kern\n", "# Author: Robert Kern\n",
"\n", "\n",
"Z = np.random.randint(0,5,(10,3))\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", "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)" "print(U)"
] ]
}, },

View File

@ -1119,9 +1119,13 @@ print(rows)
# Author: Robert Kern # Author: Robert Kern
Z = np.random.randint(0,5,(10,3)) Z = np.random.randint(0,5,(10,3))
E = np.logical_and.reduce(Z[:,1:] == Z[:,:-1], axis=1)
U = Z[~E]
print(Z) 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) print(U)
``` ```