如何将二维列表中相同向量编码为同一值？（PyTorch/Numpy实现）

Numpy实现方法 #

直接使用np.unique()函数，指定axis=0以整行作为单位判断是否重复，同时设置return_inverse=True，该参数会返回原数组每行在唯一行集合中的索引，恰好就是需要的标记结果。

代码示例：

import numpy as np

a = np.array([[0,0], [0,0], [0,0], [0,2], [0,2], [1,0]])
_, labels = np.unique(a, axis=0, return_inverse=True)
print(labels)  # 输出: [0 0 0 1 1 2]

PyTorch实现方法 #

PyTorch的torch.unique()函数支持相同逻辑，指定dim=0按行处理重复项，return_inverse=True返回对应索引数组。

代码示例：

import torch

a = torch.tensor([[0,0], [0,0], [0,0], [0,2], [0,2], [1,0]])
_, labels = torch.unique(a, dim=0, return_inverse=True)
print(labels.numpy())  # 输出: [0 0 0 1 1 2]

内容的提问来源于stack exchange，提问作者zihao jiao