数值计算用python编写程序，求解