删除numpy数组中的以下重复项

Question

我想删除相互跟随的重复项,但不要删除整个数组中的重复项.另外,我想保持订单不变.

所以如果输入[0 0 1 3 2 2 3 3]应该是输出[0 1 3 2 3]

我找到了一种方法,itertools.groupby()但我正在寻找更快的numpy解决方案.

Answer 1

a[np.insert(np.diff(a).astype(np.bool), 0, True)]
Out[99]: array([0, 1, 3, 2, 3])

一般的想法是用来diff找到数组中两个连续元素之间的差异.然后我们只索引那些给出non-zero差异元素的东西.但由于长度 diff是1短,从而在索引之前,我们需要insert对True涉及diff阵列的开始.

说明:

In [100]: a
Out[100]: array([0, 0, 1, 3, 2, 2, 3, 3])

In [101]: diff = np.diff(a).astype(np.bool)

In [102]: diff
Out[102]: array([False,  True,  True,  True, False,  True, False], dtype=bool)

In [103]: idx = np.insert(diff, 0, True)

In [104]: idx
Out[104]: array([ True, False,  True,  True,  True, False,  True, False], dtype=bool)

In [105]: a[idx]
Out[105]: array([0, 1, 3, 2, 3])