2020-08-20

# -*- coding:utf-8 -*-
import sqlite3
import pandas as pd
import os

st_list=input("请输入您要分析财报的股票代码(以逗号分开,如601111,002241,603986):")
st_list=st_list.split(",")
db = 'stockfin.db'
conn = sqlite3.connect(db)
for st in st_list:
    pl_sql='SELECT * from pl_quarter where 代码="{}"  order by "年报日期";'.format(st)
    df=pd.read_sql_query(pl_sql, conn)
    print(df)
    df.iloc[:,5:28]=df.iloc[:,5:28]+df.iloc[:,5:28].shift(1)+df.iloc[:,5:28].shift(2)+df.iloc[:,5:28].shift(3)
    print(df)
    csv_name="A.csv"
    # 获取今日日期，并转换为字符串的形式。以此日期命名建立文件路径
    df.to_csv(os.path.join(csv_name),encoding="utf_8_sig")
    # 删除包含NaN值得任何行
    df=df.iloc[3:,1:] #取第三行第1列开始的所有行列,防止列level_0重复
    print(df)
    df.to_csv(os.path.join("b.csv"),encoding="utf_8_sig")
    df.to_sql('pl_roll_year',con=conn,if_exists ='append')     #年累计"""
conn.close()