Mailman 3 Finding a row match within a numpy array - NumPy-Discussion

Aug. 15, 2007

      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.

Thanks
Andy

Finding a row match within a numpy array

Andy Cheesman

tags

participants (6)