w3resource

NumPy: Compare two given arrays


Compare Two Arrays Element-Wise

Write a NumPy program to compare two arrays using NumPy.

Pictorial Presentation:

Python NumPy: Compare two given arrays.

Sample Solution:

Python Code:

# Importing the NumPy library with an alias 'np'
import numpy as np

# Creating NumPy arrays 'a' and 'b'
a = np.array([1, 2])
b = np.array([4, 5])

# Displaying the arrays 'a' and 'b'
print("Array a: ", a)
print("Array b: ", b)

# Comparing elements of 'a' and 'b' for greater than operation
print("a > b")
print(np.greater(a, b))

# Comparing elements of 'a' and 'b' for greater than or equal operation
print("a >= b")
print(np.greater_equal(a, b))

# Comparing elements of 'a' and 'b' for less than operation
print("a < b")
print(np.less(a, b))

# Comparing elements of 'a' and 'b' for less than or equal operation
print("a <= b")
print(np.less_equal(a, b)) 

Sample Output:

Array a:  [1 2]                                                        
Array b:  [4 5]                                                        
a > b                                                                  
[False False]                                                          
a >= b                                                                 
[False False]                                                          
a < b                                                                  
[ True  True]                                                          
a <= b                                                                 
[ True  True]

Explanation:

In the above code -

a = np.array([1, 2]): A 1D NumPy array a is created with elements [1, 2].

b = np.array([4, 5]): A 1D NumPy array b is created with elements [4, 5].
print(np.greater(a, b)): The np.greater function compares the elements of a and b element-wise and returns a boolean array with the result of the comparison a > b. In this case, the output is [False False].

print(np.greater_equal(a, b)): The np.greater_equal function compares the elements of a and b element-wise and returns a boolean array with the result of the comparison a >= b. In this case, the output is [False False].

print(np.less(a, b)): The np.less function compares the elements of a and b element-wise and returns a boolean array with the result of the comparison a < b. In this case, the output is [True True].

print(np.less_equal(a, b)): The np.less_equal function compares the elements of a and b element-wise and returns a boolean array with the result of the comparison a <= b. In this case, the output is [True True].

Python-Numpy Code Editor: