我有一组 Nx2 离散坐标numpy.ndarray
.
我想获取每个独特坐标集的计数和索引。numpy.unique http://docs.scipy.org/doc/numpy/reference/generated/numpy.unique.html正是这样做的,但是对于标量元素。
有没有一些干净的方法可以使用numpy
?
Example:
#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
EDIT:
unique count
,将给出每个唯一值的计数,即: 1[10,10]
, 2 of [12,9]
和 1 个[10,5]
。然后可以找到这些对应的值coor[unique_index]
您可以使用.count()
and .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]
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)