Numpy Slicing - 奇怪的行为

如何解决Numpy Slicing - 奇怪的行为

我在使用 Numpy 时遇到问题。基本上我只想在 for 循环中做一个简单的赋值。但奇怪的是，这不起作用。这是我的示例代码：

import numpy as np
test = np.zeros((1280,),dtype=int)
idx = 0
for i in range(32):
       test[idx:idx+40] = np.ones((40,int)*1
       idx = idx + 1
print(np.where(test==0))

我假设在循环之后 test 中的所有值都等于 1，但是我的程序的输出是 (array([ 71,72,73,...,1277,1278,1279]),) 我不明白为什么数组中仍然有 0 值。特别是例如在索引 72 处，它位于循环的第二次迭代的中间。

你能帮我吗？ 谢谢！

解决方法

当您将 idx 用于索引时，它的最大值将为 31，因此在您增加它之前。从数组索引 31 开始，您将 40 个元素设置为 1。最后一个这样的元素位于索引 31+39 处，即 70。那么索引 71 处的元素仍为 0。