[Numpy-discussion] extract elements of an array that are contained in another array?
Ning Sean
ningsean at gmail.com
Wed Jun 3 20:29:31 EDT 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.python.org/pipermail/numpy-discussion/attachments/20090603/b6897782/attachment.html>
More information about the NumPy-Discussion
mailing list