SymPy contains many number-theory-related routines: obtaining prime numbers, integer decompositions, and much more. We will show a few examples here.
To display legends using LaTeX in matplotlib, you will need an installation of LaTeX on your computer (see this chapter's Introduction).
In : from sympy import * init_printing() In : import sympy.ntheory as nt
In : nt.isprime(2011) Out: True
In : nt.nextprime(2011) Out: 2017
In : nt.prime(1000) Out: 7919
In : nt.primepi(2011) ...