Skip to content

Instantly share code, notes, and snippets.

@shotahorii
Created February 3, 2019 21:26
Show Gist options
  • Save shotahorii/dfd4d4c0ca83e5b79e37f6e41b8a8a87 to your computer and use it in GitHub Desktop.
Save shotahorii/dfd4d4c0ca83e5b79e37f6e41b8a8a87 to your computer and use it in GitHub Desktop.
Nth prime
Display the source blob
Display the rendered blob
Raw
{
"cells": [
{
"cell_type": "code",
"execution_count": 6,
"metadata": {
"collapsed": true
},
"outputs": [],
"source": [
"def is_prime(n):\n",
" if n<=0:\n",
" return False\n",
" elif n<=2:\n",
" return True\n",
" else:\n",
" i = 2\n",
" while i < n:\n",
" if n%i == 0:\n",
" return False\n",
" i += 1\n",
" return True"
]
},
{
"cell_type": "code",
"execution_count": 18,
"metadata": {
"collapsed": false
},
"outputs": [],
"source": [
"def nth_prime(n):\n",
" primes = []\n",
" i = 1\n",
" while len(primes) < n:\n",
" if is_prime(i):\n",
" primes.append(i)\n",
" i += 1\n",
" return primes[-1]"
]
},
{
"cell_type": "code",
"execution_count": 27,
"metadata": {
"collapsed": false
},
"outputs": [
{
"data": {
"text/plain": [
"True"
]
},
"execution_count": 27,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"is_prime(13)"
]
},
{
"cell_type": "code",
"execution_count": 28,
"metadata": {
"collapsed": false
},
"outputs": [
{
"data": {
"text/plain": [
"43"
]
},
"execution_count": 28,
"metadata": {},
"output_type": "execute_result"
}
],
"source": [
"nth_prime(15)"
]
}
],
"metadata": {
"kernelspec": {
"display_name": "Python 3",
"language": "python",
"name": "python3"
},
"language_info": {
"codemirror_mode": {
"name": "ipython",
"version": 3
},
"file_extension": ".py",
"mimetype": "text/x-python",
"name": "python",
"nbconvert_exporter": "python",
"pygments_lexer": "ipython3",
"version": "3.5.2"
}
},
"nbformat": 4,
"nbformat_minor": 0
}
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment