宅家自学量化投资 - 建立股票数据库 (上)

路过的大兄弟好,我是菜菜的宅家汪,现在是宅家学习量化投资的73天。

最近这一个月有点松懈了哈,这个月开始想改变一下之前闭门造车式的学习,用通俗易懂的话在B站、知乎、CSDN上分享自己的所学所思所想,并用采用边做边记录的形式。那么话不多说直接进入正题。

实现功能:用tushare(一个免费开源的股票数据接口),在本地建立股票数据库。

一、问题:避免重复调取数据

首先股票数据调取过程为:1、通过接口获取数据。2、保存股票数据。

所以我的思路是创建或更新文件时,记录下文件位置和修改时间,以供下次更新数据时做为判断条件,下面来一步一步实现。

获取股票列表

首先先定义获取股票列表的函数

def get_stock_list(list_status:tuple=('L')):

    '''

    获取股票列表

    :param list_status: 上市状态:默认'L'上市、D退市、P暂停上市

    :return: 返回股票列表

    '''

    pro = ts.pro_api('xxx')

    # 列表容器

    list = []

    for s in list_status:

        data = pro.stock_basic(exchange='', list_status=s, fields='ts_code,name,area,industry,list_date')

        list.append(data)

    return pd.concat(list)


然后获取股票列表的更新日期进行判断决定是否更新,如果没找到文件则初始化股票列表。这里我是将修改日期直接命名在了文件名上

Tips:运用datatime模块、r模块、os模块

if __name__ == '__main__':

    # 获取文件所在文件夹

    files = os.listdir(r'G:\PycharmProjects\DeltaTrader\data\finance')

    for f in files:

        file_date = re.search('(^\d{4}-\d{2}-\d{2})_tss', f) # 获取文件指定修改日期

        # 判断捕获到时间

        if file_date:

            date_str = file_date.group(1)

            date = datetime.datetime.strptime(date_str, '%Y-%m-%d') # 转换为时间类型

            # 获取对应时间的星期数 和 一年中的第几周 已经当前对应的日期

            weekday = date.weekday()

            week_count = date.isocalendar()[1]

            today = datetime.datetime.today()

            # 更新日期是否不是今天,则进行更新:不是同一周 or 不是同一星期且更新日期不是星期五

            if ((weekday != today.weekday()) & (weekday != 4)) or (weekday != today.isocalendar()[1]):

                # 获取股票列表并保存和修改更新日期

                data = get_stock_list()

                data.to_csv(r'G:\PycharmProjects\DeltaTrader\data\finance\{}.csv'.format(f), index=False)

                os.rename(r'G:\PycharmProjects\DeltaTrader\data\finance\{}.csv'.format(f),

                          f'{datetime.date.today()}_tss.csv')

            break

    # 找不到文件则直接获取股票列表

    else:

        data = get_stock_list(('l', 'D'))

        data.to_csv(r'G:\PycharmProjects\DeltaTrader\data\finance\{}_tss.csv'.format(datetime.date.today()), index=False)

到这里我们就得到了更新股票数据所需的,股票代码和上市退市日期。

在下一个篇章我会继续介绍如何批量获取股票数据。 作者:阿尔法电波脑 https://www.bilibili.com/read/cv12455345/?from=readlist 出处:bilibili

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 217,185评论 6 503
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 92,652评论 3 393
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 163,524评论 0 353
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 58,339评论 1 293
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 67,387评论 6 391
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 51,287评论 1 301
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 40,130评论 3 418
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 38,985评论 0 275
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 45,420评论 1 313
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 37,617评论 3 334
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 39,779评论 1 348
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 35,477评论 5 345
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 41,088评论 3 328
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 31,716评论 0 22
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,857评论 1 269
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 47,876评论 2 370
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 44,700评论 2 354

推荐阅读更多精彩内容