Find the union of two NumPy arrays

Introduction

In data analysis and scientific computing, combining datasets is a common task. NumPy, a powerful library for numerical computations in Python, provides efficient methods for such operations. One such method is np.union1d(), which returns the unique, sorted union of two input arrays. This article explores how to use np.union1d() and its applications.

Understanding np.union1d()

The function np.union1d(ar1, ar2) computes the union of two arrays, returning a sorted array of unique elements present in either of the input arrays. It's important to note that:

• Input arrays are flattened if they are not already 1-dimensional.
• Duplicate elements are removed, ensuring uniqueness.
• The result is sorted in ascending order.

Basic Example

import numpy as np

arr1 = np.array([10, 20, 30, 40])
arr2 = np.array([20, 40, 60, 80])

union_result = np.union1d(arr1, arr2)
print("Union of arr1 and arr2:", union_result)

[10 20 30 40 60 80]

Handling Multi-dimensional Arrays

When dealing with multi-dimensional arrays, it's necessary to flatten them before performing the union operation. Here's how you can do it:

arr1 = np.array([[1, 2], [3, 4]])
arr2 = np.array([3, 4, 5, 6])

flattened_union = np.union1d(arr1.flatten(), arr2)
print("Union of flattened arr1 and arr2:", flattened_union)

[1 2 3 4 5 6]

Union of Multiple Arrays

To find the union of more than two arrays, you can use Python's functools.reduce() function in conjunction with np.union1d():

from functools import reduce

arrays = [np.array([1, 2, 3]), np.array([3, 4, 5]), np.array([5, 6, 7])]
union_all = reduce(np.union1d, arrays)
print("Union of all arrays:", union_all)

[1 2 3 4 5 6 7]

Conclusion

The np.union1d() function is a versatile tool in NumPy for combining datasets. By understanding its usage and applications, you can efficiently perform union operations on arrays, facilitating data analysis and manipulation tasks. Whether you're working with one-dimensional or multi-dimensional arrays, NumPy provides the functionality to handle these operations seamlessly.