numpy.unique_inverse

numpy.unique_inverse(x)

查找 x 的唯一元素和重建 x 的索引。

此函数是 Array API 兼容的替代方案。

>>> x = np.array([1, 1, 2])
>>> np.unique(x, return_inverse=True, equal_nan=False)
(array([1, 2]), array([0, 0, 1]))

参数:

xarray_like

输入数组。如果它不是一维的，它将被展平。

返回值:

outnamedtuple

包含以下内容的结果

• values - 输入数组的唯一元素。

• inverse_indices - 来自唯一元素集合的索引，用于重建 x。

参见

unique

查找数组的唯一元素。

示例

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