Find if something is also somewhere else | 找一找那里是不是也有

In today’s class we played a game: finding the numbers in a group that are also in another group. As usual, we started simple using numpy (np).   Here is the summary:

score1 = np.array([0, 1, 3, 5, 10,3])

score2 = np.array([2,3])

We use the inld function from numpy to do the matching and counting:
np.in1d(score1,score2)

score1[np.in1d(score1,score2)]

Our result is:

array([3, 3])

There is another way to do this:

np.array([item in score2 for item in score1])

array([False, False,  True, False, False,  True])

score1[np.array([item in score2 for item in score1])]

will give you exactly the same answer.   Try it.

For very large groups of numbers or words, what we’ve just learned will work just as well.

Please practice and turn in the homework.

Leave a Reply

Your email address will not be published.