import pandas as pd import matplotlib.pyplot as plt plt.rcParams['font.sans-serif'] = ['SimHei'] import tushare as ts import os import tushare as ts ts.set_token('bb00bf226ce5a2f24de40a18c62596d511508ecc4b83b7c62b17b86d') pro = ts.pro_api() def get_data(code,start,end): df=pro.daily(ts_code=code,autype='qfq',start_date=start,end_date=end) print(df) df.index = pd.to_datetime(df.trade_date) #设置把日期作为索引 #df['ma'] = 0.0 # Backtrader需要用到 #df['openinterest'] = 0.0 # Backtrader需要用到 #定义两个新的列ma和openinterest df = df[['open', 'high', 'low', 'close', 'vol']] #重新设置df取值,并返回df return df def acquire_code(): #只下载一只股票数据,且只用CSV保存 未来可以有自己的数据库 inp_code =input('600893.SH') inp_start = input('2019-01-01') inp_end = input('2020-01-01') df = get_data(inp_code,inp_start,inp_end) print(df.info()) #输出统计各列的数据量 print("—"*30) #分割线 print(df.describe()) #输出常用统计参数 df.sort_index(inplace=True) #把股票数据按照时间正序排列 path = os.path.join(os.path.join(os.getcwd(), "D:\学习\AI\project\pythonProject-tushare\data"), inp_code + "yy.csv") #os.path地址拼接,''数据地址''为文件保存路径 # path = os.path.join(os.path.join(os.getcwd(),"数据地址"),inp_code+"_30M.csv") df.to_csv(path) acquire_code() 代码修正
时间: 2023-06-18 19:08:40 浏览: 61
import pandas as pd
import matplotlib.pyplot as plt
plt.rcParams['font.sans-serif'] = ['SimHei']
import tushare as ts
import os
ts.set_token('bb00bf226ce5a2f24de40a18c62596d511508ecc4b83b7c62b17b86d')
pro = ts.pro_api()
def get_data(code,start,end):
df=pro.daily(ts_code=code,autype='qfq',start_date=start,end_date=end)
print(df)
df.index = pd.to_datetime(df.trade_date)
#设置把日期作为索引
#df['ma'] = 0.0 # Backtrader需要用到
#df['openinterest'] = 0.0 # Backtrader需要用到
#定义两个新的列ma和openinterest
df = df[['open', 'high', 'low', 'close', 'vol']]
#重新设置df取值,并返回df
return df
def acquire_code():
#只下载一只股票数据,且只用CSV保存
#未来可以有自己的数据库
inp_code =input('请输入股票代码:')
inp_start = input('请输入开始日期(如:2019-01-01):')
inp_end = input('请输入结束日期(如:2020-01-01):')
df = get_data(inp_code,inp_start,inp_end)
print(df.info())
#输出统计各列的数据量
print("-"*30)
#分割线
print(df.describe())
#输出常用统计参数
df.sort_index(inplace=True)
#把股票数据按照时间正序排列
path = os.path.join(os.path.join(os.getcwd(),"data"), inp_code + ".csv")
#os.path地址拼接,''数据地址''为文件保存路径
# path = os.path.join(os.path.join(os.getcwd(),"数据地址"),inp_code+"_30M.csv")
df.to_csv(path)
acquire_code()