For a given number, check whether the number is a prime number or not. in python code example

Example 1: how to check if a number is prime in python

import math
def prime(x):
    if x == 2 or x == 3 or x==5:
        return True
    if x == 1 or x == 4:
        return False
    num = 2
    a= math.sqrt(x)+4 
    while a > num:
        if x % num == 0:
            return False
        else:
            num = num + 1
    return True

Example 2: check if a number is prime python

n=input('Enter the number you want to check: ')
try:
    n=int(n)
except:
    print('Wrong input.')
    quit()
if n==1 or n==0:
    print('This is neither prime nor composite')
else:
    c=0
    for i in range(2,n):
        if n%i==0:
            c=c+1
    if c==0:
        print("This is a prime number")
    else:
        print('This  is a composite number.')