[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