[Numpy-discussion] extract elements of an array that are contained in another array?

Ning Sean ningsean@gmail....
Wed Jun 3 19:29:31 CDT 2009


Hi, I want to extract elements of an array (say, a) that are contained in
another array (say, b). That is, if a=array([1,1,2,3,3,4]), b=array([1,4]),
then I want array([1,1,4]).

I did the following but the speed is very slow (maybe because a is very
long):

c=array([])
for x in b:
   c=append(c,a[a==x])

any way to speed it up?

Thanks!
-Ning
-------------- next part --------------
An HTML attachment was scrubbed...
URL: http://mail.scipy.org/pipermail/numpy-discussion/attachments/20090603/b6897782/attachment.html 


More information about the Numpy-discussion mailing list