概要
numpyのarrayを使った和集合と差集合のまとめです。
nunique・appendなどのメソッドや、set型を組み合わせれば、工夫次第で実現できますが、用意された関数が既にあるようなので、それを使ってしまいましょう!
準備
In [1]: import numpy as np
In [2]: array_a = np.array([1, 2, 3, 4, 5])
In [3]: array_b = np.array([1, 3, 5, 7, 9])
和集合(Union)
np.union1d(A, B)は、np.array型の新しい集合A∪B = {x|x∈A 又は x∈B}
を返します。
In [4]: np.union1d(array_a, array_b)
Out[4]: array([1, 2, 3, 4, 5, 7, 9])
差集合(Difference set)
np.setdiff1d(A, B)は、np.array型の新しい集合A\B = {x|x∈A かつ x∉B}
を返します。
In [5]: np.setdiff1d(array_a, array_b)
Out[5]: array([2, 4])