気ままに実装する機械学習

機械学習に興味のある大学院生によるブログです. 機械学習以外のトピック多めです.

MENU

numpyで集合演算をするための備忘録

研究で集合演算を使うことがあるのですが、すぐに使い方を忘れてしまうため、自分用の備忘録として記事にしたいと思います.

積集合

 A \cap B

使い方

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

np.intersect1d(a, b)

結果

array([2, 3])

和集合

 A \cup B

使い方

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

np.union1d(a, b)

結果

array([1, 2, 3, 4, 5])

差集合

 A \backslash B

使い方

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

np.setdiff1d(a, b)

結果

array([1, 4])

排他的論理和

 A \quad xor \quad B

使い方

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

np.setxor1d(a, b)

結果

array([1, 4, 5])