Python module to plot cosmic-ray flux