Merge pull request #34 from ibah/master
generalizing the solution to the 'extract rows with unequal values'
This commit is contained in:
commit
f4d6512cef
@ -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)"
|
||||||
]
|
]
|
||||||
},
|
},
|
||||||
|
@ -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)
|
||||||
```
|
```
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user