Python3:IndexErrorの解消方法

Question

python初心者です。エラーの対処法を教えていただきたいです。

import numpy as np
def pred(array,color):
    image=array
    length=len(image)
    for i in range(length**2):
        num1=(i)/3
        num2=(i)%3
        image[num1][num2]=(image[num1][num2]+color)/2
    return image

a=np.array([[[  0,128,  0],[  0,  0,128],[  0,  0,128]],
          [[  0,128,  0],[  0,  0,  0],[  0,  0,  0]],
          [[  0,128,  0],[128,  0,  0],[128,  0,  0]]], dtype=np.uint8)

b=np.array([100,100,100],dtype=np.uint8)

print(pred(a,b))

以上のコードを入れたとき次のようなエラーが出てしまいました。

  image[num1][num2]=(image[num1][num2]+color)/2
IndexError: only integers, slices (`:`), ellipsis (`...`), numpy.newaxis (`None`) and integer or boolean arrays are valid indices

このエラーの意味するところとどのように書き換えを行えばエラーが解消されるでしょうか。回答いただけると幸いです。

Answer 1

num1=(i)/3をnum1=(i)//3に修正。
python3では、整数同士の割り算に/を使うと、浮動小数点数が返るため。

Comments

[nanahiryu]

エラー解消されました！ありがとうございます！

[nanahiryu]

そんな仕様があるんですね。もう少し勉強勧めたいと思います。