import statsmodels.formula.api as smf
import statistics
import numpy as np
import pandas as pd
class G():
pass
g = G()
def init(ContextInfo):
print('测试')
g.gpc=['002601.SZ']
g.gpc.append('000300.SH')
def handlebar(ContextInfo):
global mean_turnover
dtime = ContextInfo.get_market_data_ex(['close'], stock_code =\
g.gpc,period = '1d',start_time='20230325',end_time='20230928', count=-1,\
dividend_type = 'follow',fill_data=True,subscribe=True)
if len(dtime['000300.SH'])>65:
for i in dtime.keys():
return_signal(dtime, 'close',i)
return
def return_signal(data:dict,field:str,code:str):
_index=data[list(data.keys())[0]].index.tolist()
_columns=list(data.keys())
df=pd.DataFrame(index=_index,columns=_columns)
for i in _columns:
df[i]=data[i][field]
df1=data[code][-61:-1]
df=df.rename(columns={'000300.SH':'SH'})
df=df.rename(columns={code:'SZ'})
test_df = df.iloc[-61:-1]
test_df = (test_df/test_df.iloc[0]).apply(np.log)
test_df['idd'] = np.array(range(60))/252.0
test_df=pd.DataFrame(test_df)
print(test_df.columns)
return
上面的代码运行后,如下图,不是应该只显示['SZ''SH''IDD']怎么还会打印出['002601.SZ']这一项
 |