a3b39f9e6e7e1b33a3eb9919923d939430b60b86,QUANTAXIS/QAFetch/QATdx.py,,QA_fetch_get_stock_latest,#Any#Any#Any#,157

Before Change

    code = [code] if isinstance(code, str) else code
    api = TdxHq_API(multithread=True)
    with api.connect(ip, port):
        data = pd.DataFrame()
        for item in code:
            market_code = __select_market_code(item)
            __data = api.to_df(api.get_security_bars(
                9, market_code, item, 0, 1))
            __data["code"] = item
            data = data.append(__data)
        data["date"] = data["datetime"].apply(lambda x: x[0:10])
        data["date_stamp"] = data["date"].apply(
            lambda x: QA_util_date_stamp(x))
        data["date"] = pd.to_datetime(data["date"])
        data.set_index("date", drop=False, inplace=True)
        data["date"] = data["date"].apply(lambda x: str(x)[0:10])
        return data.drop(["year", "month", "day", "hour",
                          "minute", "datetime"], axis=1)

After Change

    code = [code] if isinstance(code, str) else code
    api = TdxHq_API(multithread=True)
    with api.connect(ip, port):
        data = pd.concat([api.to_df(api.get_security_bars(
            9, __select_market_code(item), item, 0, 1)).assign(code=item) for item in code], axis=0)
        return data\
            .assign(date=pd.to_datetime(data["datetime"]
                                        .apply(lambda x: x[0:10])), date_stamp=data["datetime"]
                    .apply(lambda x: QA_util_date_stamp(str(x[0:10]))))\

Instances