以几何级数从 numpy 数组中删除值

Question

我想以增量方式删除元素，例如

arr = [1,2,3,4,5,6,7,8,9,10]
# want to delete first element, second element, forth element
# 1,2,4,8,16,32,62.... till length of list
# For above example output will be
arr = [3,5,6,7,9,10]

arr 是numpy数组。 有什么方法可以定义自定义 position 来删除元素。 现在我能想到的就是写一个自定义的 function

import pandas as pd
import math
from numpy import random

arr = random.randint(100, size=(10))
get_index = [2**i for i in range(math.log(len(arr)))]
arr = numpy.delete(arr, index)

如果有一些 numpy function 可以做到这一点，那就太好了

Answer 1

您可以使用不需要的元素的索引创建一个set ：

from math import log, ceil

arr = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]

del_idx = {2 ** i for i in range(ceil(log(len(arr), 2)))}
print(del_idx)  # {8, 1, 2, 4}

res = [x for i, x in enumerate(arr, start=1) if i not in del_idx]
print(res)  # [3, 5, 6, 7, 9, 10]

索引 go 到log(len(arr), 2) 。

然后你可以enumerate你的数组并忽略你不想要的索引。

---

使用numpy你可以这样做：

import numpy as np
from math import log, ceil

arr = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
del_idx = [2 ** i - 1 for i in range(ceil(log(len(arr), 2)))]
arr = np.delete(arr, del_idx)

Answer 2

import numpy as np
arr = #numpy array
# Filtering indices with no power of 2 numbers
idx = [i for (i,n) in enumerate(arr) if not n & (n - 1) == 0]
arr = arr[idx]

或者你可以简单地这样做（单行版）：

arr = arr[(arr & (arr - 1) != 0)]

其中 & 是按位运算符。 从这里提取的过滤器。