Python numpy.unique的2D等效项

我在Nx2

numpy.ndarray

中有一组离散坐标

我想得到这些唯一坐标集的计数和索引。确实如此，但对于标量元素

使用

numpy

有什么干净的方法可以做到这一点吗

例如：

#input
coor = np.array([[10,10],[12,9],[10,5],[12,9]]) 
#output
unique_count = np.array([1,2,1])
unique_index = np.array([0,1,2]) #1 could also be 3

编辑：

唯一计数

将给出每个唯一值的计数，即：1个

[10,10]

，2个

[12,9]

和1个

[10,5]

。然后，您可以使用

.count（）

和

.index（）

列表的方法找到这些值对应的值

coor = np.array([[10, 10], [12, 9], [10, 5], [12, 9]])
coor_tuple = [tuple(x) for x in coor]
unique_coor = sorted(set(coor_tuple), key=lambda x: coor_tuple.index(x))
unique_count = [coor_tuple.count(x) for x in unique_coor]
unique_index = [coor_tuple.index(x) for x in unique_coor]

---

你能更详细地解释一下唯一计数和唯一索引的内容吗？我不明白。@Ohumeronen因为我想要的是坐标对的索引，而不是我自己开始理解的每个X/Y坐标的计数……我尝试了一下，但得到了以下输出

（数组（[5,9,10,12]）、数组（[5,3,0,2]）、数组（[1,2,3,2]）

，=>3

10

，2

12

和

9

和1

5

。我认为它可能会在计算唯一值之前将数组展平，正如索引所示。是的，你是对的，请参阅我的编辑。我不认为这是做你想做的事情的最快的方法，但它是有效的。