view python/prime.py @ 253:8593b0384d3e

add a function for computing primes
author Jeff Hammel <jhammel@mozilla.com>
date Mon, 10 Dec 2012 11:08:21 -0800
parents 05728a85e19b
children b5a59c3e4421
line wrap: on
line source

#!/usr/bin/env python

def prime(number):
    half = int(number / 2)
    for i in range(2, half):
        if not number % i:
            return False
    return True

def primes(n):
    return [i for i in range(2,n)
            if not [True for j in range(2,1 + i/2)
                    if not i%j]]


if __name__ == '__main__':
    import sys
    for arg in sys.argv[1:]:
        print prime(int(arg))