用n替换numpy数组中的多个元素

Question

在给定的numpy数组X ：

X = array([1,2,3,4,5,6,7,8,9,10])

我想分别用单个元素-1替换索引(2, 3)和(7, 8) ，如：

X = array([1,2,-1,5,6,7,-1,10])

换句话说，我用原始值替换原始数组的索引(2, 3)和(7,8)处的值。

问题是：它周围是否有一种numpy-ish方式（即没有for循环和使用python列表）？ 谢谢。

注意：这不等于用另一个元素就地替换单个元素。 它用“奇异”值替换多个值。 谢谢。

Answer 1

使用numpy.delete的解决方案，类似于@pault，但更高效，因为它使用纯粹的numpy索引。 但是，由于这种有效的索引，这意味着您不能将锯齿状数组作为索引传递

设定

a = np.array([1,2,3,4,5,6,7,8,9,10])
idx = np.stack([[2, 3], [7, 8]])

---

a[idx] = -1
np.delete(a, idx[:, 1:])

array([ 1,  2, -1,  5,  6,  7, -1, 10])

Answer 2

我不确定这是否可以一步完成，但这是使用np.delete的方法：

import numpy as np
from operator import itemgetter

X = np.array(range(1,11))
to_replace = [[2,3], [7,8]]


X[list(map(itemgetter(0), to_replace))] = -1
X = np.delete(X, list(map(lambda x: x[1:], to_replace)))
print(X)
#[ 1  2 -1  5  6  7 -1 10]

首先，我们用-1替换每对中的第一个元素。 然后我们删除剩余的元素。

Answer 3

试试np.put ：

np.put(X, [2,3,7,8], [-1,0]) # `0` can be changed to anything that's not in the array
print(X[X!=0]) # whatever You put as an number in `put`

所以基本上使用put来做索引的值，然后删除零值。

或者像@khan所说，可以做一些超出范围的事情：

np.put(X, [2,3,7,8], [-1,np.max(X)+1])
print(X[X!=X.max()])

所有输出：

[ 1  2 -1  5  6  7 -1 10]