[Numpy-discussion] another little index puzzle
Robert Kern
robert.kern at gmail.com
Mon Jan 21 08:50:53 EST 2013
On Mon, Jan 21, 2013 at 2:41 PM, Neal Becker <ndbecker2 at gmail.com> wrote:
> I have an array to be used for indexing. It is 2d, where the rows are all the
> permutations of some numbers. So:
>
> array([[-2, -2, -2],
> [-2, -2, -1],
> [-2, -2, 0],
> [-2, -2, 1],
> [-2, -2, 2],
> ...
> [ 2, 1, 2],
> [ 2, 2, -2],
> [ 2, 2, -1],
> [ 2, 2, 0],
> [ 2, 2, 1],
> [ 2, 2, 2]])
>
> Here the array is 125x3
>
> I want to select all the rows of the array in which all the 3 elements are
> equal, so I can remove them. So for example, the 1st and last row.
all_equal_mask = np.logical_and.reduce(arr[:,1:] == arr[:,:-1], axis=1)
some_unequal = arr[~all_equal_mask]
--
Robert Kern
More information about the NumPy-Discussion
mailing list