Подсчет уникальных значений в массиве NumPy

Из этого туториала Вы узнаете, как подсчитать количество уникальных значений в массиве NumPy.

Подсчет уникальных значений в массиве NumPy с помощью функции numpy.unique()

Чтобы подсчитать количество вхождений каждого уникального элемента в массиве numpy, мы можем использовать функцию numpy.unique(). Он принимает массив в качестве входного аргумента и возвращает все уникальные элементы внутри массива в порядке возрастания. Мы можем указать для параметра return_counts значение True, чтобы также получить количество повторений каждого элемента внутри массива. См. Следующий пример кода.

import numpy as np

array = np.array([1,1,1,2,3,4,4,4])

unique, counts = np.unique(array, return_counts=True)

result = np.column_stack((unique, counts)) 
print (result)

Выход:

[[1 3]
 [2 1]
 [3 1]
 [4 3]]

Сначала мы создали массив NumPy с функцией np.array(). Затем мы сохранили все уникальные элементы array внутри unique массива и их соответствующее количество вхождений внутри массива counts с помощью функции np.unique(). Затем мы заархивировали два одномерных массива unique и counts в один 2D-массив result с помощью функции np.column_stack().