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