年明けの嵐が怖いので、大好き素数を求めて気を落ち着けるしか。

#!/usr/local/bin/python



def get_prime_list(num):

  known_prime = [2]

  current_integer = 3

  while len(known_prime) < num:

    for indy in known_prime:

      if current_integer % indy == 0:

        current_integer = current_integer + 2

        break

    else:

      known_prime.append(current_integer)

      current_integer = current_integer + 2

  return known_prime



import sys

integer = int(sys.argv[1])

print integer,"th prime is",get_prime_list(integer)[integer-1]

current_integer = current_integer + 2 の部分がちょっと気が利いているような気がしていたんだけど、実行速度は +1 でもほとんどかわらないんじゃん。