从 numpy 数组中的字符串中按索引删除元素

Question

我有以下带有字符串的示例数组：

 ['100000000' '101010100' '110101010' '111111110']

我希望能够同时通过数组中每个字符串中的索引删除一些元素。 例如，如果我要删除索引为 6 和 8 的元素，我应该会收到以下结果：

 ['1000000' '1010110' '1101000' '1111110']

到目前为止我所有的尝试都失败了，可能是因为字符串是不可变的，但我不确定我是否必须转换，如果是的话 - 转换成什么以及如何转换。

Answer 1

import numpy as np

a = np.array(['100000000', '101010100', '110101010', '111111110'])

list(map(lambda s: "".join([c for i, c in enumerate(str(s)) if i not in {5, 7}]), a))

退货：

['1000000', '1010110', '1101000', '1111110']

Answer 2

另一种方法是将a转换为单个字符的二维数组，屏蔽掉不需要的值，然后再转换回字符串的一维数组。

import numpy as np

a = np.array(['100000000', '101010100', '110101010', '111111110'])

b = a.view('U1').reshape(*a.shape, -1)

mask = np.ones(b.shape[-1], dtype=bool)
mask[[5, 7],] = False

b = b[:, mask].reshape(-1).view(f'U{b.shape[-1] - (~mask).sum()}')