python return prime factors code example

Example: prime factorization python

import math


def primeFactors(n):
    # no of even divisibility
    while n % 2 == 0:
        print(2)
        n = n / 2
    # n reduces to become odd
    for i in range(3, int(math.sqrt(n)) + 1, 2):
        # while i divides n
        while n % i == 0:
            print(i)
            n = n / i
    # if n is a prime
    if n > 2:
        print(n)


primeFactors(256)