时间戳

首页 > 解决方案 > 有没有办法从python中的数组中删除相似的（数字）元素

python - 有没有办法从python中的数组中删除相似的（数字）元素

问题描述

我有一个函数可以生成这样的数组：

[ 14  48  81 111 112 113 114 148 179 213 247 279 311 313 314 344 345 346]

它对应于曲线与 x 轴相交的数据值。由于数据不完善，它会产生误报，其中我的输出数组的元素彼此非常接近，例如 [111 112 113 114]。我需要从该数组中删除误报，但仍保留显示误报的初始正数。基本上我需要我的函数来产生和排列更像

[ 14  48  81 112 148 179 213 247 279 313 345]

来自不完美数据的误报已被删除。

标签： pythonarraysnumpy

解决方案

这是一种可能的方法：

arr = [14, 48, 81, 111, 112, 113, 114, 148, 179, 213, 247, 279, 311, 313, 314, 344, 345, 346]

def filter_arr(arr, offset):
    filtered_nums = set()
    for num in sorted(arr):
        # Check if there are any "similar" numbers already found
        if any(num+x in filtered_nums for x in range(-offset, offset+1)):
            continue
        else:
            filtered_nums.add(num)
    return list(sorted(filtered_nums))

然后，您可以使用您认为最有意义的任何偏移量应用过滤。

filter_arr(arr, offset=5)  
Output:  [14, 48, 81, 111, 148, 179, 213, 247, 279, 311, 344]

推荐阅读