python 计算股票保本卖出价格

给定买入股票价格，股票数量，券商佣金，计算保本卖出价格，若手续费不满5按5元收取。

输出的第一个值为保本最低卖出价格（这里加了0.01，保证卖出时不会因为四舍而亏损）。

def fun(p,n,f=0.00025):
    #印花税 0.001
    #过户费0.00002
    #佣金不满5收5元
    #p=买入价格,n=买入n股,f=佣金费率，默认单边万2.5 
    #输出 保本卖出价格，买入手续费，卖出手续费
    
    #买入花费金额
    buy=p*n+p*n*f+p*n*0.00002
    #买入手续费
    buy_fee=buy-p*n
    #判断手续费是否满5
    if buy_fee>=5:
        pass
    else:
        buy_fee=5
    #保本最低卖出价格
    s=buy/(n-n*f-n*0.00002-n*0.001)
    sell_fee=n*f*s+s*n*0.00002+s*n*0.001
    if sell_fee>=5:
        pass
    else :
        s=(buy+5)/(n-n*0.00002-n*0.001)
        
    return round(s,2)+0.01,round(buy_fee,2),round(sell_fee,2)