python编写股票公式