3
4

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?

More than 5 years have passed since last update.

np.arrayが他のnp.arrayに含まれるか判定

Posted at

例えば、こんな感じになってた時、


import numpy as np
a = np.array([1,2,4,2,5])
b = np.array([2,2,4,5])
c = np.array([1,3]) 

baに含まれてるか、caに含まれてるか、判定したいときのやり方です

1. np.isin を使う

np.isin(arr1, arr2)とすると、arr1の各要素がarr2に含まれているかのTrue/Falseのnp.arrayが帰ってきます
これを使って、


sum(np.isin(b, a)) == len(b) # returns True
sum(np.isin(c, a)) == len(c) # returns False

のように判定することができます。
np.isinnp.in1dに変更しても全く同じ結果になります。

2. np.intersect1dを使う

np.intersect1d(arr1, arr2)で、arr1arr2の共通部分を返してくれます。
デフォルトでは同じ要素は一つのみ返ってくるので、np.intersect1d(a, b)array([2, 4, 5])を返すことになります。
それゆえ、以下のようにして比較できます


len(np.intersect1d(a, b)) == len(np.unique(b)) # returns True
len(np.intersect1d(a, c)) == len(np.unique(c)) # returns False

3. 単純にsetにして包含関係を見る

うだうだnumpyでやってきましたが、単純にこれでいいですよね


set(b) <= set(a) # returns True
set(c) <= set(a) # returns False

他に良いやり方があったら教えてください

3
4
0

Register as a new user and use Qiita more conveniently

  1. You get articles that match your needs
  2. You can efficiently read back useful information
  3. You can use dark theme
What you can do with signing up
3
4

Delete article

Deleted articles cannot be recovered.

Draft of this article would be also deleted.

Are you sure you want to delete this article?