python - 如何更改 numpy 数组中的数据 block

Question

我在 Python 中有一个大型 numpy 一维数据数组，并希望将 x (500) 到 y (520) 的条目更改为等于 1。我可以使用 for 循环，但是否存在一种更简洁、更快的 numpy 方法？

对于范围内的 x (500,520)
numpyArray[x] = 1.

这是可以使用的 for 循环，但似乎 numpy 中可能有一个我遗漏的函数 - 我宁愿不使用 numpy 提供的屏蔽数组

Answer 1

您可以使用 [] 访问一系列元素:

import numpy as np

a = np.ones((10))
print(a) # Original array
# [ 1.  1.  1.  1.  1.  1.  1.  1.  1.  1.]
startindex = 2
endindex = 4
a[startindex:endindex] = 0

print(a) # modified array
# [ 1.  1.  0.  0.  1.  1.  1.  1.  1.  1.]