cupy.unique#

cupy.unique(ar, return_index=False, return_inverse=False, return_counts=False, axis=None, *, equal_nan=True)[源代码][源代码]#

查找数组中的唯一元素。

返回数组中已排序的唯一元素。除了唯一元素之外，还有三个可选输出：

  • 输入数组的索引，这些索引给出了唯一值

  • 重建输入数组的唯一数组的索引

  • 输入数组中每个唯一值出现的次数

参数:

  • ar (array_like) – 输入数组。如果它还不是一维的，将会被展平。

  • return_index (bool, optional) – 如果为 True，还会返回 ar 的索引（沿着指定的轴，如果提供，或者在展平的数组中），这些索引导致唯一的数组。

  • return_inverse (bool, optional) – 如果为 True，还会返回唯一数组的索引（如果提供了指定的轴），这些索引可以用来重建 ar。

  • return_counts (bool, optional) – 如果为真，还返回每个唯一项在 ar 中出现的次数。

  • axis (int or None, optional) – 要操作的轴。如果为 None，ar 将被展平。如果为整数，则由给定轴索引的子数组将被展平并视为 1-D 数组的元素，其维度为给定轴，详见注释。默认值为 None。

  • equal_nan (bool, optional) – 如果为真，将返回数组中的多个 NaN 值合并为一个。

返回:

如果没有可选输出，它返回排序后的唯一值的 cupy.ndarray。否则，它返回一个元组，其中包含排序后的唯一值及以下内容。* 唯一值在原始数组中首次出现的索引。仅当 return_index 为 True 时提供。* 从唯一数组重构原始数组的索引。仅当 return_inverse 为 True 时提供。* 每个唯一值在原始数组中出现的次数。仅当 return_counts 为 True 时提供。

返回类型:

cupy.ndarray or tuple

备注

当指定一个轴时，由该轴索引的子数组将被排序。这是通过将指定的轴移动到数组的第一维度（将轴移动到第一维度以保持其他轴的顺序），然后按C顺序展平子数组来完成的。

警告

此功能可能会同步设备。

参见

numpy.unique()