[Numpy-discussion] finding nonzero elements in list
per freem
perfreem at gmail.com
Mon Oct 12 10:18:44 EDT 2009
hi all,
i'm trying to find nonzero elements in an array, as follows:
a = array([[1, 0],
[1, 1],
[1, 1],
[0, 1]])
i want to find all elements that are [1,1]. i tried: nonzero(a ==
[1,0]) but i cannot interpret the output. the output i get is:
(array([0, 0, 1, 2]), array([0, 1, 0, 0]))
i simply want to find the indices of the elements that equal [1,0].
how can i do this? thanks.
More information about the NumPy-Discussion
mailing list