Как остановить печать на Python более чем на одну строку?

Question

Почему мой код выводит несколько строк без необходимости?

n = int(input("Enter a number to find if it is prime:  "))

def is_prime():
    for i in range(2, n):
        if n % i == 0:
            print("Not prime")
            break
        else:
            print("The number {} is prime".format(n))

is_prime()

Если я введу число 2, например, при запуске программы, оно напечатает:

the number 2 is prime

the number 2 is prime

the number 2 is prime

Нужно напечатать только одну строку, так почему же это так?

Answer 1

Ваш else находится в неправильном положении. У вас есть это на if, но вы действительно хотите это на for.

Возможно, он не очень известен, но вы можете иметь else для for -циклов, и он будет выполняться, если во время циклов не было выполнено break.

n = int(input("Enter a number to find if it is prime:  "))

def is_prime():
    for i in range(2, n):
        if n % i == 0:
            print("Not prime")
            break
    else:
        print("The number {} is prime".format(n))

is_prime()