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]: aOut[100]: array([0, 0, 1, 3, 2, 2, 3, 3])In [101]: diff = np.diff(a).astype(np.bool)In [102]: diffOut[102]: array([False, True, True, True, False, True, False], dtype=bool)In [103]: idx = np.insert(diff, 0, True)In [104]: idxOut[104]: array([ True, False, True, True, True, False, True, False], dtype=bool)In [105]: a[idx]Out[105]: array([0, 1, 3, 2, 3])
