np.unique()

Функція повертає унікальні елементи ( рядки, стовпці) масиву і супутню інформацію.

Синтаксис за замовчуваням

 
np.unique(ar, return_index=False, return_inverse=False, return_counts=False, axis=None)

Приклади роботи при різних значеннях параметрів

Для одномірного масиву перетворює мультимножину його значень в відсортовану множину значень

   
A=np.array([1, 1, 2, 2, 3, 3,4,5,6,6])
np.unique(A)
array([1, 2, 3, 4, 5, 6])

Те саме для двовимірного масиву

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

Повертає унікальні рядки масиву

   
A = np.array([[1, 0, 0], [1, 0, 0], [2, 7, 4]])
np.unique(A,axis=0)
array([[1, 0, 0],
       [2, 7, 4]]) 
       

Повертає унікальні стовпці масиву

   
A = np.array([[1, 0, 0], [1, 0, 0], [2, 7, 4]])
np.unique(A,axis=1)
   
array([[0, 0, 1],
       [0, 0, 1],
       [4, 7, 2]])  

Те саме і додатково індекси ( в масиві унікальних елементів) унікальних елементів

   
A=np.array([1, 1, 2, 2, 3, 3,4,5,6,6])
_, indices = np.unique(A, return_index=True)
array([0, 2, 4, 6, 7, 8], dtype=int64)

В комбінації з осями, для двовимірного масиву повертає індекси унікальних рядків і стовпців

Додатково - кількість унікальних елементів

 
A = np.array([[1, 0, 0], [1, 0, 0], [2, 7, 4]])
u, counts = np.unique(A, return_counts=True)
u
array([0, 1, 2, 4, 7])

counts
array([4, 2, 1, 1, 1], dtype=int64)