Ну хорошо, моя ошибка, что не уточнил.
Уточню даже больше - это 3 питон. Так будет лучше?
Update:
А вообще, agata, избыточно на каждую подобную фигню писать по программе. Для такого есть мат. пакеты, типа octave, matlab, matematika(??? как то так) + спец. языки программирования, типа R, с готовыми интерфейсами. Лучше проинвестируйте 50 денег в покупку книжки, тем более, что заказчик - математик.