numpy.unique_inverse

numpy.unique_inverse(x)

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

此函数是 Array API 与以下内容兼容的替代方案：

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

但返回一个命名元组以方便访问每个输出。

参数：

x类数组

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

返回值：

out命名元组

包含以下内容的结果：

• 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])