Welcome, guest | Sign In | My Account | Store | Cart

Find the nth prime in python (Python recipe) by Captain DeadBones
ActiveState Code (http://code.activestate.com/recipes/578403/)

This is a nth prime number generator in python that I put together for an article I wrote How To Find Prime Numbers In Python

Python, 27 lines
Download
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
import sys
import math 

def is_prime(num):
	for j in range(2,int(math.sqrt(num)+1)):
		if (num % j) == 0: 
			return False
	return True

def main(argv):

	if (len(sys.argv) != 2):
		sys.exit('Usage: prime_numbers4.py <nth_prime>')

	i = 0
	num = 2
	nth = int(sys.argv[1])
		
	while i < nth:
		if is_prime(num): 
			i += 1
			if i == nth:
				print 'The ' + str(nth) + ' prime number is: ' + str(num)
		num += 1 
	
if __name__ == "__main__":
	main(sys.argv[1:])
Tags: primes, prime_generator, prime_number