神秘的 IndexError - 非常基本的问题（Python）

Question

我的一些代码有问题，我觉得我必须在这里遗漏一些基本的东西。一个与我遇到相同错误的简单示例如下：

from numpy import array,zeros
x = array([1,2,3])
f = zeros(len(x))

for i in x:
    f[i] = x[i] + 1

回溯如下：

---------------------------------------------------------------------------
IndexError                                Traceback (most recent call last)
C:\WINDOWS\system32\<ipython-input-5-6b6b88f30156> in <module>()
      1 for i in x:
----> 2     f[i] = x[i] + 1
      3 

IndexError: index out of bounds

这让我困惑太久了，但我似乎看不出这里有什么问题？有人可以帮忙吗？

Answer 1

在这个循环中：

for i in x:
    f[i] = x[i] + 1

i 取值 1、2 和 3。x[i] 不是你想的那样。 i 已经包含数组 x 的单元格的内容。由于数组的索引从 0 开始，因此在尝试获取索引为 3 的元素（将是第 4 个元素）时执行IndexError。

你可能想要这样的东西：

for i in range(len(x)):
    f[i] = x[i] + 1

也可以这样写：

for i, v in enumerate(x):
    f[i] = v + 1

Answer 2

当您执行for i in some_list 时，i 指的是该列表的元素，而不是它们的索引。例如：

In [1]: for i in [3, 2, 1]:
   ...:     print i
   ...:     
3
2
1

但是，您随后使用 i 作为索引。

您在x 上使用i 进行迭代，因此i 采用1、2 和3 的值。但是3 对于长度为 3 的数组来说是一个太大的索引。最后一个索引是2，因为在 Python 中索引从 0 开始。