cupy.unique#

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

查找数组中的唯一元素。

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

  • 给出唯一值的输入数组的索引

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

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

参数:

  • ar (类数组) – 输入数组。如果它不是一维数组，则会被展平。

  • return_index (bool, 可选) – 如果为 True，则还会返回导致生成唯一数组的 ar 的索引（如果提供了指定轴，则沿指定轴；否则在展平后的数组中）。

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

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

  • axis (int 或 None, 可选) – 要操作的轴。如果为 None，则 ar 会被展平。如果为整数，则由给定轴索引的子数组将被展平，并作为与给定轴维度相同的一维数组的元素处理，详情请参阅备注。默认值为 None。

  • equal_nan (bool, 可选) – 如果为 True，则返回数组中的多个 NaN 值将被合并为一个。

返回:

如果没有可选输出，则返回已排序唯一值的 cupy.ndarray。否则，它返回包含已排序唯一值和以下内容的元组。

  • 原始数组中唯一值首次出现的索引。仅在 return_index 为 True 时提供。

  • 从唯一数组重建原始数组的索引。仅在 return_inverse 为 True 时提供。

  • 原始数组中每个唯一值出现的次数。仅在 return_counts 为 True 时提供。

返回类型:

cupy.ndarray 或 tuple

备注

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

警告

此函数可能会同步设备。

另请参阅

numpy.unique()