Pythonで計算結果cをリストに格納できません。

Question

Pythonで計算結果cをリストgz[i][j]に格納gz[i][j]=cできなくて困っています。
cの計算や代入はprint(i,j,c)で確認できています。

cもgzもココが初出で、前後のコードでの使用 や　定義もしていません。

    for i in range(len(x)):
        for j in range(len(x)):
            c = (a1*x[i]**3)+(a2*x[i]**2*y[j])
             print(i,j,c)
             gz[i][j]=c

エラーメッセージ

 gz[i][j]=c
TypeError: 'int' object does not support item assignment

下記の様にダイレクトに格納する場合に、
事前にgzを２次元配列と定義するコードは必要なのでしょうか？

 gz[i][j] = (a1*x[i]**3)+(a2*x[i]**2*y[j])

Comments

[ozwk]

そのコードより前にgzに代入してるところはありませんか(引数としてでも)

[shiracamus]

gz が 1次元配列だとしたら、同様のエラーになります。
>>> gz = [0]
>>> gz[0][0] = 0
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: 'int' object does not support item assignment

[shiracamus]

> 事前にgzを２次元配列と定義するコードは必要なのでしょうか？

必要です。

gz = [[0 for i in range(len(x))] for j in range(len(x))]

Answer 1

プログラム全体が書かれていないので
a1やa2は勝手に起きましたが
以下のコードで
gzに格納することができます．
格納するgzを先に定義する必要があると思います．

import numpy as np

a1 = 5
a2 = 6
x = [0,1,2,3,4]
y = [0,1,2,3,4]
gz= np.zeros((5,5))
for i in range(len(x)):
    for j in range(len(x)):
        c = (a1*x[i]**3)+(a2*x[i]**2*y[j])
        print(i,j,c)
        gz[i][j]=c
print(gz)

実行結果
0 0 0
0 1 0
0 2 0
0 3 0
0 4 0
1 0 5
1 1 11
1 2 17
1 3 23
1 4 29
2 0 40
2 1 64
2 2 88
2 3 112
2 4 136
3 0 135
3 1 189
3 2 243
3 3 297
3 4 351
4 0 320
4 1 416
4 2 512
4 3 608
4 4 704
[[ 0. 0. 0. 0. 0.]
[ 5. 11. 17. 23. 29.]
[ 40. 64. 88. 112. 136.]
[135. 189. 243. 297. 351.]
[320. 416. 512. 608. 704.]]

Answer 2

gz[i][j]=c

TypeError: 'int' object does not support item assignment

このエラーメッセージは、gzがint型であるのに、配列のようにインデックスを付けて使おうとしている事を示しています。

質問のプログラムの最初に、以下のようなコードでgzを2次元配列として宣言すれば解決します。

    gz=[[0]*3]*3

質問のプログラムでは、gz[i][j]のiとjが0からlen(x)-1までの範囲で変化しますから、上記のコードの３をlen(x)の値に置き換えてください。
len(x)が512なら、gz=[[0]*512]*512 というぐあいに。

Comments

[icemanstanding]

ありがとうございました。

[shiracamus]

> gz=[[0]*3]*3

間違った初期化方法ですよ。

>>> gz=[[0]*3]*3
>>> gz
[[0, 0, 0], [0, 0, 0], [0, 0, 0]]
>>> gz[0][0] = 999
>>> gz
[[999, 0, 0], [999, 0, 0], [999, 0, 0]]