python, get the n-th prime