基本说明
本实例采用数据集是a股历史分钟数据
arctic 项目是老牌对冲基金man group自研后开源到github,底层是基于mongodb,但是速度快了10倍以上,数据压缩率高于30%
本实例是入门教程,前提是有mongodb的基础知识,以及如何读取存入数据。
连接到mongodb
连接到本地mongodb,或者连接到mongodb server
# connect to local mongodb
store = Arctic('localhost')
# connect to mongodb server
store = Arctic('mongodb://.....')
新建arctic数据库
这里我们用的是2021年a股1分钟数据
store.initialize_library('Ashare_Minute_2021')
连接arctic数据库
lib = store['Ashare_Minute_2021']
综上所述,三条代码完成数据库的创建和连接,非常简洁高效。
数据写入
这里我们需要自定义写入函数:
由于采用本地csv数据导入,所以遍历下文件夹所有股票,用arctic 自带的lib.write命令完成csv文件导入到数据库操作。
def write_data(lib):
work_dir = '/Volumes/Seagate Backup Plus Drive/BaiduNetdiskDownload/2021/'
for file in os.listdir(work_dir):
file_path = os.path.join(work_dir, file)