素数判定：NameError ２のときだけおかしい

Question

###実現したいこと
２以上ｘ以下の整数ｘを読み込んでそれが素数か判定したいです。
また、素数でないときは一番小さい約数を表示します。
###問題点
2のときだけ下記のようなエラーがでます。たぶんflgという名前が定義されていないという意味だと思いますがどう定義したらいいのかわかりません。

2
Traceback (most recent call last):
  File "kadai02b.py", line 13, in <module>
    if flg:
NameError: name 'flg' is not defined

###コード

x=int(input())
i=0
num=0
for i in range(2,x):
    #素数ではない
    if x%i==0:
        flg=False
        num=i
        break
    else:
        flg=True
if flg:
    print("prime")
else:
    print(num)

Answer 1

2つの引数(start, stop)を指定したときのrange関数の動作は、「start以上、stop未満の連番を生成する」というものなので、startとstopが同じ値(今回だと、どちらも2)の場合には空になり、for文に含まれるif文は一度も実行されません。

すると、変数flgが定義されないまま、for文に続くif文でflgの値が使われるため、NameErrorが発生します。

解決策としては、for文より手前で変数flgにTrueを代入してしまうことです。その結果、for文に含まれるif文のelse以下は不要になります。

x = int(input())
flg = True
num = 0

for i in range(2, x):
    if x % i == 0:
        flg = False
        num = i
        break

print("prime" if flg else num)

Comments

[langhtorn]

こうなってしまう理由がよくわかりました。丁寧にありがとうございます。

Answer 2

２のときはfor文はスキップされてしまうんで、flgが定義されないため、それがエラーになってますね

これをどうにかするならflg=Trueをコードの最初に入れておきましょう

Comments

[langhtorn]

定義の仕方をありがとうございます。