numpy.unique_inverse#

numpy.unique_inverse(x)[源代码]#

x 的唯一元素和重建 x 的索引.

此函数是与 Array API 兼容的替代方法,用于:

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

但返回一个 namedtuple,以便更轻松地访问每个输出.

备注

此函数目前始终返回排序后的结果,但是,这可能会在任何 NumPy 次要版本中更改.

参数:
xarray_like

输入数组.如果它还不是 1-D 的,它将被展平.

返回:
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])