NumPy: Count the frequency of unique values in numpy array

NumPy: Array Object Exercise-94 with Solution

Write a NumPy program to count the frequency of distinct values in a NumPy array.

Pictorial Presentation:

Sample Solution:

Python Code:

# Importing the NumPy library and aliasing it as 'np'
import numpy as np

# Creating a NumPy array 'a' containing integers
a = np.array([10, 10, 20, 10, 20, 20, 20, 30, 30, 50, 40, 40])

# Printing a message indicating the original array will be displayed
print("Original array:")

# Printing the original array 'a' with its elements
print(a)

# Finding unique elements and their counts in the array 'a' using np.unique() with return_counts=True
unique_elements, counts_elements = np.unique(a, return_counts=True)

# Printing a message indicating the frequency of unique values in the array
print("Frequency of unique values of the said array:")

# Creating a NumPy array from the unique elements and their respective counts
# Converting the resulting arrays into a 2D NumPy array using np.asarray()
result = np.asarray((unique_elements, counts_elements))

# Printing the array containing unique elements and their frequencies
print(result) 

Sample Output:

Original array:                                                        
[10 10 20 10 20 20 20 30 30 50 40 40]                                  
Frequency of unique values of the said array:                          
[[10 20 30 40 50]                                                      
 [ 3  4  2  2  1]]

Explanation:

In the above code –

• a = np.array(...): Create a NumPy array 'a' containing the given integer values.
• np.unique(a, return_counts=True): Find the unique elements in the array 'a' and their counts using the np.unique function. The return_counts parameter is set to True, so the function returns two arrays: one containing the unique elements and another containing the corresponding counts of those unique elements.
• unique_elements, counts_elements = ...: Assign the returned unique elements and their counts to the variables unique_elements and counts_elements, respectively.
• np.asarray((unique_elements, counts_elements)): Combine the unique elements and their counts into a single 2D NumPy array.
• print(...): Print the resulting 2D NumPy array.

Python-Numpy Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Write a NumPy program to get the magnitude of a vector in numpy.
Next: Write a NumPy program to check whether the numpy array is empty or not.