Python for Cosmology