IndexError：索引 1 超出轴 0 的范围，大小为 1（Python，NumPy）

Question

我正在使用 for 循环遍历 NumPy zeros 数组中的索引，并分配一些值为 0.5 的索引。目前，我的代码返回错误信息：

IndexError：索引 1 超出轴 0 的范围，大小为 1

下面是我的代码的简化版本，它重现了错误。

import numpy as np

Z = np.zeros((1560, 1560))
linestart = {1: [175], 2: [865]}
noycuts = 2

cutno = int(0)
for i in range(noycuts):
    cutno = cutno + 1
    xstart = linestart[cutno]
    ystart = 0
    for j in range(1560):
        Z[xstart][ystart] = 0.5
        ystart = ystart + 1

我检查了来自具有相同错误代码的人的问题，尽管这些问题似乎源于最初调用数组的方式；我不认为这是我的问题。

任何人都可以看到我的代码中导致错误消息的缺陷吗？

我希望我已经提供了足够的信息。

提前致谢。

Answer 1

编辑：

我原来的答案是：

替换

Z[xstart][ystart] = 0.5

与

Z[xstart, ystart] = 0.5

但实际上，问题在于您的 xstart 是一个数组。保留您的原始代码，但替换

linestart = {1: [175], 2: [865]}

与

linestart = {1: 175, 2: 865}

或者，更好：

linestart = [175, 865]

Answer 2

使用linestart = {1: [175], 2: [865]}，您可以定义一个包含单个条目的列表的字典。我相信您实际上希望字典包含整数。 ystart 也应该从零开始。以下是否符合您的要求：

import numpy as np

Z = np.zeros((1560, 1560))
linestart = {1: 175, 2: 865}
noycuts = 2

cutno = 0
for i in range(noycuts):
    cutno += 1
    xstart = linestart[cutno]
    ystart = 0
    for j in range(1560):
        Z[xstart][ystart] = 0.5
        ystart = ystart + 1

还要考虑以下哪个是较短的版本：

for cutno,xstart in linestart.items():
    for ystart in range(Z.shape[1]):
        Z[xstart][ystart] = 0.5