Python 素数を判定する関数を作って整数Ｎ以上の素数を15個表示したい

Question

まず最初に素数を判定する関数を作りました。
キーボードから整数を入力してその数以上の素数を順番に15個表示させるプログラムを作りたいです。
Trueの場合に表示とカウントをしようと考えましたが上手くいきませんでした。

# 素数判定の関数です
def isPrimeNumber( n ):
    if n < 2:
        return False
    elif n == 2:
        print(n)
        return True
    for i in range(2,n):
        if n % i == 0:
            return False
    else:
        print(n)
        return True
# 本体です
N = int(input())
count = 0
while True:
    isPrimeNumber(N)
    if count >15:
        break
    count +=1
    N +=1

Comments

[maisumakun]

「上手くいきませんでした。」とありますが、上記のプログラムを実行すると、どうなるのでしょうか？

[magepi]

例えば50と入力すると50～65の間にある素数を出力してしまいました。

Answer 1

この場合、関数の中でprint()するのではなく
while文内で、関数を実行した際にTrueが返ってきた場合にcount += 1とし同時にprint()しましょう。
if文と組み合わせる事で、isPrimeNumber関数でTrueが返ってきた場合にif文内のコードが実行されます。

# 素数判定の関数です
def isPrimeNumber( n ):
	if n < 2:
		return False
	elif n == 2:
		return True
	for i in range(2,n):
		if n % i == 0:
			return False
	else:
		return True
# 本体です
N = int(input())
count = 0
while count < 15:
	if isPrimeNumber(N):
		count += 1
		print(N, count)
	N +=1

Comments

[magepi]

ありがとうございます関数の使い方を理解できていませんでした。if文と組み合わせる発想が出てこなくてとても勉強になりました！

[nto]

if文では比較演算子等(a > b や a == True)を使用しない場合には、その値がTrueかFalseかを判別し、Trueだった場合に実行するという性質があります。
その為、isPrimeNumber()を実行しTrueが返ってきた場合にはコードが実行されるという仕組みです。
もちろんわかりやすくif isPrimeNumber(n) == True:も同じ条件式となります。
maisumakunさんがご回答されている様に、せっかく関数内でTrueやFalseを戻り値としてreturnしているのですから、戻り値を上手に活用しましょう。

Answer 2

Trueの場合に表示とカウントをしようと考えましたが

その割に、isPrimeNumber(N)とだけして、判定結果を捨ててしまっています。

Comments

[magepi]

判定して出てきた戻り値を上手く使えていないようでした。ありがとうございました。