numpy.unique_inverse#

numpy.unique_inverse(x)[源]#

查找 x 的唯一元素以及用于重构 x 的索引。

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

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

但返回一个命名元组(namedtuple)以便更方便地访问每个输出。

注意

此函数目前总是返回一个排序后的结果,然而,这在任何 NumPy 次要版本中都可能发生变化。

参数:
x类数组(array_like)

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

返回:
out命名元组(namedtuple)

包含以下结果:

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