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])