[Numpy-discussion] Finding a row match within a numpy array
Stefan van der Walt
stefan at sun.ac.za
Sat Aug 18 06:21:07 EDT 2007
On Tue, Aug 14, 2007 at 11:53:03AM +0100, Andy Cheesman wrote:
> Dear nice people
>
> I'm trying to match a row (b) within a large numpy array (a). My most
> successful attempt is below
>
> hit = equal(b, a)
> total_hits = add.reduce(hit, 1)
> max_hit = argmax(total_hits, 0)
> answer = a[max_hit]
>
> where ...
> a = array([[ 0, 1, 2, 3],
> [ 4, 5, 6, 7],
> [ 8, 9, 10, 11],
> [12, 13, 14, 15]])
>
> b = array([8, 9, 10, 11])
>
>
>
> I was wondering if people could suggest a possible more efficient route
> as there seems to be numerous steps.
Another way to do it:
a[N.apply_along_axis(N.all,1,a==b)]
Cheers
Stéfan
More information about the NumPy-Discussion
mailing list