实战：创建股票数据库

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

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

1.1创建example模块

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

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)

运行结果：

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)

运行 成功获取到数据：

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

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

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

# 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)

运行：成功添加索引名称

转载请注明：虚坏叔叔 » Python量化交易实战-11创建股票数据库开发实战