如何更改numpy数组中的数据块

Question

我在Python中有一个大的numpy 1维数据数组，并希望条目x（500）到y（520）更改为等于1.我可以使用for循环但是有一个整洁的，这样做的速度更快numpy？

for x in range(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.]