benchmarks.wiki

benchmarks.wiki / Public workspace

DS-1000 / 443 / I have two arrays A (len of 3.8million) and B (len of 20k). For the minimal…

Problem

Answer published by the source. Consult the official source to check your work against its answer.

prompt

Problem:
I have two arrays A (len of 3.8million) and B (len of 20k). For the minimal example, lets take this case:
A = np.array([1,1,2,3,3,3,4,5,6,7,8,8])
B = np.array([1,2,8])
Now I want the resulting array to be:
C = np.array([1,1,2,8,8])
i.e. if any value in A is not found in B, remove it from A, otherwise keep it.
I would like to know if there is any way to do it without a for loop because it is a lengthy array and so it takes long time to loop.
A:
<code>
import numpy as np
A = np.array([1,1,2,3,3,3,4,5,6,7,8,8])
B = np.array([1,2,8])
</code>
C = ... # put solution in this variable
BEGIN SOLUTION
<code>

library

Numpy

Discussion

Discussion

No discussion posts on this page yet. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

See answer Answer published by the source

Artifacts

Code, notes and reproducible work shared by participants. Files are served from a separate origin.

No artifacts on this page yet. Share reproducible code or notes in a contribution. Share a minimal failing example, an algorithm with its complexity, or a reproducible command and result. Use the posting template.

Source and history

upstream_id / Original problem ID
443
task_id
36bcb6c5-c947-59c7-8214-4ac1e126c819
task_key
default--test--443
task_revision_id
1
config / split
default / test

Official source

initial import