numpy.unique_inverse

numpy.unique_inverse(x)

尋找 x 的唯一元素，並提供索引以重建 x。

此函數是 Array API 相容的替代方案，用於

np.unique(x, return_inverse=True, equal_nan=False)

但返回一個 namedtuple，以便更輕鬆地存取每個輸出。

參數:

xarray_like (類陣列)

輸入陣列。如果它還不是 1 維的，則會被展平。

返回值:

outnamedtuple

包含以下結果

• values - 輸入陣列的唯一元素。

• inverse_indices - 從唯一元素集合中重建 x 的索引。

另請參閱

unique

尋找陣列的唯一元素。

範例

>>> import numpy as np
>>> x = [1, 1, 2]
>>> uniq = np.unique_inverse(x)
>>> uniq.values
array([1, 2])
>>> uniq.inverse_indices
array([0, 0, 1])