用python语言编写一个函数

Question

1.编写一个函数，可以利用tushare同时获取多支股票数据：通过tushare获取到不同公司的盈利数据，并对roe大于15，net_profit_ratio>30的股票选取出来。

Answer 1

使用tushare获取股票在一定时期的盈利能力函数是get_profit_data（），获取数据后用pandas截取数据即可，写成函数代码如下：

def profit_capability(roe,npr):
    df= ts.get_profit_data(2021, 2)
    #df.to_excel('stock2021q2.xlsx',index=False)
    #df = pd.read_excel('stock2021q2.xlsx')
    res = df[(df['roe'] > roe) & (df['net_profit_ratio'] > npr)]
    return res
roe=15
npr=30
r=profit_capability(roe,npr)
print(r)
r.to_excel('stock2021q2.xlsx',index=False)

如有帮助，请点采纳。