在使用choice数据库时,常常会做一些重复性的操作,比如我会经常对标的基金的行业相关性进行计算,所以把这些操作封装了几个函数。
由于我常使用申万行业指数,所以函数就命名ShenWan了。
具体代码如下所示:
import pandas as pd
import datetime
today=datetime.date.today()
today.strftime("%Y-%m-%d")
start_date='2021-04-01'
end_date=today.strftime("%Y-%m-%d")
# 计算申万一级行业指数与各基金指数之间的关系
shenwan_code=""
def ShenWan(shenwan_code,start_date=start_date,end_date=end_date):
shenwan=c.csd(shenwan_code,"CLOSE",start_date,end_date,"period=1,adjustflag=1,curtype=1,order=1,market=CNSESH")
d1=shenwan.Data
name=c.css(shenwan_code,"CODE,NAME")
name=pd.DataFrame(name.Data)
for i in d1.keys():
d1[i]=d1[i][0]
d1=pd.DataFrame(d1)
d1.columns=name.loc[1]
d1.index=shenwan.Dates
d1.to_csv('./d1.csv',index=False)
fund=''
def Fund(fund,start_