Is there a Python package people recommend that has code for these kinds of calculations?