Мне нравится подход, когда математика "на коленке" сначала прикидывается в мат. редакторе - Wolfram или Mathcad, а потом уже более сложные расчеты (если нужно) развиваются в Python. Сам я целиком за Mathcad, т.к. им профессионально занимаюсь. Его функционал почти не изменился за 20 лет, так что можете старыми версиями пользоваться, если они под рукой. Или скачать бесплатную редакцию
тут (правда в ней много всего выключено, напр., вся символьная математика).