[Tutor] Finding a row match within a numpy array

Andy Cheesman Andy.cheesman at bristol.ac.uk
Mon Aug 13 13:36:23 CEST 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


More information about the Tutor mailing list