SymPy
SymPyライブラリを使うと、素数を簡単に扱うことができます。
Google Colaboratoryでは、SymPy がすでにインストールされていますが、ローカル環境では下記のコマンドでSymPyをインストールする必要があります。
[Google Colaboratory]
1 | pip install sympy |
素数を求める
一定範囲の素数を求める場合は、sieve.primerangeを使います。
[Google Colaboratory]
1 | from sympy import sieve |
[実行結果]
[2, 3, 5, 7, 11, 13, 17, 19, 23, 29]
素数かどうかを判定する
ある数値が素数かどうかを判定する場合は、isprimeを使います。
[Google Colaboratory]
1 | from sympy import isprime |
[実行結果]
True