File size: 1,326 Bytes
7ec53ba |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 |
import math
import choose
import pandas as pd
import os
import datetime
from pymongo import MongoClient
client = MongoClient(
"mongodb://wth000:wth000@43.159.47.250:27017/dbname?authSource=wth000")
db = client["wth000"]
collection=db[f"策略买入ST股票('000', '001', '002', '600', '601', '603', '605')"]
collection.drop()
moneyused = 0.9 # 设置资金利用率
# 获取当前.py文件的绝对路径
file_path = os.path.abspath(__file__)
# 获取当前.py文件所在目录的路径
dir_path = os.path.dirname(file_path)
# 获取当前.py文件所在目录的上两级目录的路径
dir_path = os.path.dirname(os.path.dirname(dir_path))
path = os.path.join(
dir_path, f"股票('000', '001', '002', '600', '601', '603', '605')指标周期30交易细节.csv")
df = pd.read_csv(path)
df = df[["代码", "日期"]]
for idx, row in df.iterrows():
code = row['代码']
date = row['日期']
dfst = pd.DataFrame(list(
db[f"聚宽ST股票('000', '001', '002', '600', '601', '603', '605')"].find({"代码": code, "日期": date})))
# dfst = pd.DataFrame(list(
# db[f"股票除息除权('000', '001', '002', '600', '601', '603', '605')"].find({"代码": code, "除权日": date})))
print(code, date, dfst)
if not dfst.empty:
collection.insert_many(dfst.to_dict("records"))
|