首页 > 解决方案 > 删除数组中的连续元素

问题描述

给定一个带有排序整数的数组。我想删除数组中的连续数字。例子:

import numpy as np
array = np.array([1,2,3,6,9,13,14,15,20,25])

Expected output:
[3,6,9,13,15,20,25]

标签: pythonnumpy

解决方案


开始了:

mask_array = np.diff(array, append=False)==1 #
if len(mask_array)>1: 
    mask_array[-1] = mask_array[-2]
array[~mask_array]

输出:

array([ 3,  6,  9, 15, 20, 25])

推荐阅读