python - 我的 numpy 数组总是以零结尾？

Question

我想我错过了什么地方。我使用两个 for 循环(x 和 y)和一个基于 x,y 位置的函数填充了一个 numpy 数组。唯一的问题是无论数组的大小如何，数组的值总是以零结尾。

thetamap = numpy.zeros(36, dtype=float)
thetamap.shape = (6, 6)
for y in range(0,5):
    for x in range(0,5):
        thetamap[x][y] =  x+y
print thetamap

Answer 1

range(0, 5) 产生 0, 1, 2, 3, 4。端点总是被省略。您只需要 range(6)。

更好的是，使用 NumPy 的强大功能 将数组放在一行中:

thetamap = np.arange(6) + np.arange(6)[:,None]

这会产生一个行向量和一个列向量，然后使用 NumPy 广播将它们加在一起形成一个矩阵。