python - 私のnumpy配列は常にゼロで終わりますか?

Question

私はどこかで何かを逃したと思います。2 つの 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 の素晴らしい機能を使用して、配列を 1 行で作成します。

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

これにより、行ベクトルと列ベクトルが作成され、NumPy ブロードキャストを使用してそれらが加算され、行列が作成されます。