numpy.unique

numpy.unique(ar, return_index=False, return_inverse=False, return_counts=False, axis=None) [source]

Нахождение уникальных элементов массива.

Возвращает отсортированные уникальные элементы массива. Помимо уникальных элементов, доступны три дополнительных выходных значения:

• индексы входного массива, соответствующие уникальным значениям
• индексы массива уникальных элементов, восстанавливающие входной массив
• число появлений каждого уникального значения в входном массиве

Параметры:

ar : array_like Входной массив. Если axis не указано, он будет сглажен, если он не является одномерным. return_index : bool, optional Если True, также возвращает индексы ar (по указанной оси, если задана, или в сглаженном массиве), которые приводят к массиву уникальных элементов. return_inverse : bool, optional Если True, также возвращает индексы массива уникальных элементов (для указанной оси, если задана), которые могут быть использованы для восстановления ar. return_counts : bool, optional Если True, также возвращает количество появлений каждого уникального элемента в ar. Новое в версии 1.9.0. axis : int or None, optional Ось для обработки. Если None, ar будет сглажен. Если целое число, подмассивы, индексированные данной осью, будут сглажены и обработаны как элементы одномерного массива с размерностью данной оси, см. заметки для получения более подробной информации. Массивы объектов или структурированные массивы, содержащие объекты, не поддерживаются, если используется параметр axis. Значение по умолчанию — None. Новое в версии 1.13.0.

Возвращает:

unique : ndarray Отсортированные уникальные значения. unique_indices : ndarray, optional Индексы первых появлений уникальных значений в исходном массиве. Возвращаются только если return_index равно True. unique_inverse : ndarray, optional Индексы для восстановления исходного массива из массива уникальных элементов. Возвращаются только если return_inverse равно True. unique_counts : ndarray, optional Количество появлений каждого из уникальных значений в исходном массиве. Возвращаются только если return_counts равно True. Новое в версии 1.9.0.

См. также

numpy.lib.arraysetops

Модуль с рядом других функций для выполнения операций над множествами в массивах.

Примечания

Когда указана ось, подмассивы, индексированные этой осью, сортируются. Это делается путем преобразования указанной оси в первую размерность массива, а затем сглаживания подмассивов в порядке C. Сглаженные подмассивы затем рассматриваются как структурированный тип, и каждый элемент получает метку, в результате чего мы получаем одномерный массив структурированных типов, которые могут обрабатываться так же, как и любой другой одномерный массив. В результате сглаженные подмассивы сортируются в лексикографическом порядке, начиная с первого элемента.

Примеры

>>> np.unique([1, 1, 2, 2, 3, 3])
array([1, 2, 3])
>>> a = np.array([[1, 1], [2, 3]])
>>> np.unique(a)
array([1, 2, 3])

Возвращение уникальных строк 2D-массива

>>> a = np.array([[1, 0, 0], [1, 0, 0], [2, 3, 4]])
>>> np.unique(a, axis=0)
array([[1, 0, 0], [2, 3, 4]])

Возвращение индексов исходного массива, дающих уникальные значения:

>>> a = np.array(['a', 'b', 'b', 'c', 'a'])
>>> u, indices = np.unique(a, return_index=True)
>>> u
array(['a', 'b', 'c'],
       dtype='|S1')
>>> indices
array([0, 1, 3])
>>> a[indices]
array(['a', 'b', 'c'],
       dtype='|S1')

Восстановление исходного массива из уникальных значений:

>>> a = np.array([1, 2, 6, 4, 2, 3, 2])
>>> u, indices = np.unique(a, return_inverse=True)
>>> u
array([1, 2, 3, 4, 6])
>>> indices
array([0, 1, 4, 3, 1, 2, 1])
>>> u[indices]
array([1, 2, 6, 4, 2, 3, 2])