这个是虚幻私塾的主站。牢记域名: [xuhss.com](拼音首字母)。首页可以搜索或: 虚幻账号

Python量化交易实战-11创建股票数据库开发实战

auto-trade 虚幻 139℃ 0评论

实战:创建股票数据库

大家好,上节课我们实现了股票行情数据模块的构建,这节课我们来尝试调用一下上节课封装好的方法。

一、尝试调用stock.py里面的方法

1.1创建example模块

20210605205024 - Python量化交易实战-11创建股票数据库开发实战

创建一个Example文件夹,并且创建一个Stock.py的文件用于调用之前封装的方法。

20210605205035 - Python量化交易实战-11创建股票数据库开发实战

1.2 stock脚本

import sys,os
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))
sys.path.append(BASE_DIR)

import Data.Stock as st

# from Data import Stock
#调用一直股票的行情数据
data = st.get_single_price(stock_code='000001.XSHE', timefrequency='daily', startdate='2021-02-01', enddate='2021-03-03')

#存入csv中
st.export_data(data=data, filename='000001.XSHE', type='Price')
print(data)

运行结果:

20210605202721 - Python量化交易实战-11创建股票数据库开发实战

1.3 获取csv数据

在stock.py中定义一个通用的公共函数

import pandas as pd
    #从csv中获取数据
def get_csv_data(code, type):
    fileroot = 'C:/Users/huang/Desktop/XuhssQuant/Data/' + type +'/' + code + '.csv'
    return pd.read_csv()

然后在Example中调用这个函数:

#从csv中获取数据
data = st.get_csv_data(code='000001.XSHE',type='Price')
print(data)

运行 成功获取到数据:

20210605203810 - Python量化交易实战-11创建股票数据库开发实战

但是第一列的数据是"unnamed",怎么解决这个问题呢?这个需要从源头解决,在将data存到csv里面的时候,就需要对这一列进行命名。

如何操作呢,这里需要用到对索引重命名的方法。

20210605204144 - Python量化交易实战-11创建股票数据库开发实战

接着需要将刚才注释掉的代码解封:

# from Data import Stock
#调用一直股票的行情数据
data = st.get_single_price(stock_code='000001.XSHE', timefrequency='daily', startdate='2021-02-01', enddate='2021-03-03')

#存入csv中
st.export_data(data=data, filename='000001.XSHE', type='Price')
print(data)

#从csv中获取数据
data = st.get_csv_data(code='000001.XSHE',type='Price')
print(data)

运行:成功添加索引名称

20210605204513 - Python量化交易实战-11创建股票数据库开发实战

转载请注明:虚幻私塾 » Python量化交易实战-11创建股票数据库开发实战

喜欢 (1)

您必须 登录 才能发表评论!