PythonのNumPyライブラリには、配列の和集合を求めるための関数numpy.union1d
があります。この関数は、2つの入力配列のいずれかに存在する値のユニークでソートされた配列を返します。
numpy.union1dの基本的な使用方法
numpy.union1d
関数の基本的な使用方法は以下の通りです。
import numpy as np
# 2つの配列を定義
ar1 = np.array([-1, 0, 1])
ar2 = np.array([-2, 0, 2])
# 和集合を求める
result = np.union1d(ar1, ar2)
print(result) # 出力: array([-2, -1, 0, 1, 2])
この例では、2つの配列ar1
とar2
の和集合を求め、結果を表示しています。
複数の配列の和集合を求める
2つ以上の配列の和集合を求めるには、functools.reduce
関数と組み合わせて使用します。
from functools import reduce
# 複数の配列を定義
arrays = [np.array([1, 3, 4, 3]), np.array([3, 1, 2, 1]), np.array([6, 3, 4, 2])]
# 和集合を求める
result = reduce(np.union1d, arrays)
print(result) # 出力: array([1, 2, 3, 4, 6])
この例では、reduce
関数を使用して複数の配列の和集合を求めています。
以上が、PythonとNumPyを使用して配列の和集合を求める方法になります。この機能を活用することで、配列操作をより効率的に行うことができます。.