FAQ > 金融建模 > 第三方交互 > Python

Q:pyTSL接口中如何传递DataFrame类型数据执行    

  • A:提供以下两种方式的范例:
    方式一:pyTSL提供方法DataFrameToTSArray,可以把dataframe转成TS array风格的数据结构。
    方式二:天软和python都能支持json串,在交互过程中,可以通过json串作为数据转换的桥梁,解决编码不一致及数据类型不匹配等的问题。
    交互接口:FAQ:天软pyTSL接口
    使用案例:
    范例01:DataFrameToTSArray把DataFrame类型数据转为TS array
    Python代码:获取指定代码列表行情信息

    import pandas as pd
    import pyTSL
    stk=pd.DataFrame({"stock":["SZ000001","SZ000002","SZ000016"],"date":["2022-04-01","2022-04-01","2022-04-01"]})
    # DataFrame转为TS array
    stk=pyTSL.DataFrameToTSArray(stk)

    # 交互
    c = pyTSL.Client("user", "password", "tsl.tinysoft.com.cn", 443) #user为天软账号,password 为账号密码
    c.login()
    test = '''
    function get_data(data);
    begin
       t:=select *,
                specall(array(close(),vol(),StockZf3()),array(pn_stock():["stock"],pn_date():StrToDate(["date"]))) as "info"
          from data end;
       t[:,array("close","vol","soockzf")]:=t[:,"info"];
       deletefield(t,"info");
       return t;
    end;
    '''
    r = c.call('get_data',stk,code=test).value()
    # 可直接返回DataFrame类型数据
    print(pd.DataFrame(r))


    返回结果:

    范例02:把DataFrame格式数据转为json串
    Python代码:获取指定代码列表行情信息

    import pandas as pd
    import pyTSL
    stk=pd.DataFrame({"stock":["SZ000001","SZ000002","SZ000016"],"date":["2022-04-01","2022-04-01","2022-04-01"]})
    # dataframe转json串传参
    stk=stk.to_json(orient='records')

    # 交互
    c = pyTSL.Client("user", "password", "tsl.tinysoft.com.cn", 443) #user为天软账号,password 为账号密码
    c.login()
    test = '''
    function get_data(stk);
    begin
        //json串转天软数据类型
        data:=importjsonstring(stk);

       t:=select *,
                specall(array(close(),vol(),StockZf3()),array(pn_stock():["stock"],pn_date():StrToDate(["date"]))) as "info"
          from data end;
       t[:,array("close","vol","soockzf")]:=t[:,"info"];
       deletefield(t,"info");
       return t;
    end;
    '''
    r = c.call('get_data',stk,code=test).value()
    # 可直接返回DataFrame类型数据
    print(pd.DataFrame(r))


    返回结果: