1.获取数据
import pandas as pd
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['SimHei']
import tushare as ts
import os
def get_data(code,start='1990-1-1',end='2021-1-1'):
df=ts.get_k_data(code,autype='qfq',start=start,end=end)
# df = ts.get_k_data(code, autype='qfq', ktype="30",start=start, end=end)
print(df)
df.index=pd.to_datetime(df.date)
df['ma']=0.0 #Backtrader需要用到
df['openinterest'] = 0.0 # Backtrader需要用到
df=df[['open','high','low','close','volume','openinterest',"ma"]]
return df
def acquire_code(): #只下载一只股票数据,且只用CSV保存 未来可以有自己的数据库
inp_code =input("请输入股票代码:\n")
inp_start = input("请输入开始时间:\n")
inp_end = input("请输入结束时间:\n")
df = get_data(inp_code,inp_start,inp_end)
print(df.info())
print("—"*30)
print(df.describe())
path = os.path.join(os.path.join(os.getcwd(),"数据地址"),inp_code+".csv")
# path = os.path.join(os.path.join(os.getcwd(),"数据地址"),inp_code+"_30M.csv")
df.to_csv(path )
acquire_code()
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】博客园社区专享云产品让利特惠,阿里云新客6.5折上折
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步