GetGrowth.py 文件源码

def getGrowth(cursor):
    for i in range(1992, 2017+1):
        for j in range(1, 4+1):
            try:
                print(i, j)
                df = ts.get_growth_data(i, j)

                # ?????
                df = df.fillna(0)

                dfLen = len(df)
                # print(dfLen)
                uuidList = []  # ??uuid
                yearList = []  # ????
                quarterList = []  # ????
                for l in range(0, dfLen):
                    uuidList.append(uuid.uuid1())
                    yearList.append(str(i))
                    quarterList.append(str(j))
                df['uuid'] = uuidList
                df['year'] = yearList
                df['quarter'] = quarterList

                for k in range(0, dfLen):
                    df2 = df[k:k+1]

                    cursor.execute("insert into stock_growth(uuid, code, name, mbrg, nprg, nav, "
                               "targ, epsg, seg, year, quarter) "
                               "values(:uuid, :code, :name, :mbrg, :nprg, :nav, "
                               ":targ, :epsg, :seg,  :year, :quarter)",
                               (str(list(df2['uuid'])[0]), str(list(df2['code'])[0]), str(list(df2['name'])[0]), round(float(df2['mbrg']), 4),
                                round(float(df2['nprg']), 4), round(float(df2['nav']), 4),
                                round(float(df2['targ']), 4), round(float(df2['epsg']), 4), round(float(df2['seg']), 4),
                                str(list(df2['year'])[0]), str(list(df2['quarter'])[0])) )
                cursor.execute("commit")
            except Exception:
                pass