fromdatetimeimportdatetimefromdatetimeimporttimedeltadefdaterange(start,end):# -----make list of "date"---------------------------
forninrange((end-start).days):yieldstart+timedelta(n)#----------------------------------------------------
start=datetime.strptime(str(day_start),'%Y-%m-%d').date()end=datetime.strptime(str(day_end),'%Y-%m-%d').date()list_days=[iforiindaterange(start,end)]
ここでの start と end はtatetime関数といい、日付専用の関数です。
足したり引いたりすることは出来ません。
importpandasimportnumpyasnpimportmatplotlib.pyplotaspltimportcsvfromdatetimeimportdatetimefromdatetimeimporttimedeltadefget_stock_df(code,day_start,day_end):# (Stock Code, start y-m-d, end y-m-d)
# -----colmn name------------------------------------
# 日付、始値、高値、安値、終値、出来高、調整後終値
#Date, Open, High, Low, Close, Volume, Adjusted Close
#----------------------------------------------------
print("<<< Start stock data acquisition >>>")print("code:{0}, from {1} to {2}".format(code,day_start,day_end))defget_stock_df_under20(code,day_start,day_end):# (Stock Code, start y-m-d, end y-m-d)
# -----source: Yahoo Finance----------------------------
# Up to 20 data can be displayed on the site at one time
#-------------------------------------------------------
sy,sm,sd=str(day_start).split('-')ey,em,ed=str(day_end).split('-')url="https://info.finance.yahoo.co.jp/history/?code={0}&sy={1}&sm={2}&sd={3}&ey={4}&em={5}&ed={6}&tm=d".format(code,sy,sm,sd,ey,em,ed)list_df=pandas.read_html(url,header=0)foriinrange(len(list_df)):iflist_df[i].columns[0]=="日付":df=list_df[i]returndf.iloc[::-1]#-------------------------------------------------------
defdaterange(start,end):# -----make list of "date"---------------------------
forninrange((end-start).days):yieldstart+timedelta(n)#----------------------------------------------------
start=datetime.strptime(str(day_start),'%Y-%m-%d').date()end=datetime.strptime(str(day_end),'%Y-%m-%d').date()list_days=[iforiindaterange(start,end)]pages=len(list_days)//20mod=len(list_days)%20ifmod==0:pages=pages-1mod=20start=datetime.strptime(str(day_start),'%Y-%m-%d').date()end=datetime.strptime(str(day_end),'%Y-%m-%d').date()df_main=get_stock_df_under20(code,list_days[0],list_days[mod-1])forpinrange(pages):df=get_stock_df_under20(code,list_days[20*p+mod],list_days[20*(p+1)+mod-1])df_main=pandas.concat([df_main,df])print("<<< Completed >>> ({0}days)".format(len(df_main)))returndf_maindefgraphing(f,dt):# データのパラメータ
N=len(f)# サンプル数
dt=1# サンプリング間隔
t=np.arange(0,N*dt,dt)# 時間軸
freq=np.linspace(0,1.0/dt,N)# 周波数軸
# 高速フーリエ変換
F=np.fft.fft(f)# 振幅スペクトルを計算
Amp=np.abs(F)# グラフ表示
plt.figure(figsize=(14,6))plt.rcParams['font.family']='Times New Roman'plt.rcParams['font.size']=17plt.subplot(121)plt.plot(t,f,label='f(n)')plt.xlabel("Time",fontsize=20)plt.ylabel("Signal",fontsize=20)plt.grid()leg=plt.legend(loc=1,fontsize=25)leg.get_frame().set_alpha(1)plt.subplot(122)plt.plot(freq,Amp,label='|F(k)|')plt.xlabel('Frequency',fontsize=20)plt.ylabel('Amplitude',fontsize=20)plt.ylim(0,1000)plt.grid()leg=plt.legend(loc=1,fontsize=25)leg.get_frame().set_alpha(1)plt.show()nikkei="998407.O"panasonic="6752.T"sony="6758.T"# code info "https://info.finance.yahoo.co.jp/history/?code="
df=get_stock_df(panasonic,"2016-5-23","2019-5-23")#graphing(df["終値"],1)
csv=df.to_csv('./panasonic.csv',encoding='utf_8_sig')
