【収支】MATLAB、Pythonで株価予測 その53【シミュレーション⑤】

バックナンバーはこちら。
https://www.simulationroom999.com/blog/stock-predict-matlabpython-backnumber/

はじめに
登場人物
極大値と極小値の特定する(Python版)
動作結果
これからの話
まとめ

はじめに

前回は、極大値と極小値の特定のMATLABコードを作成した。
一部、最初に極小値が来ることを想定しているコードになっていることが問題点としてあるが、
最初に買付をする想定なので、むしろ今回のコードの方が都合が良いという判断をした。

今回は、これのPython(Numpy)版を作成する。

登場人物

博識フクロウのフクさん

イラストACにて公開の「kino_k」さんのイラストを使用しています。
https://www.ac-illust.com/main/profile.php?id=iKciwKA9&area=1

エンジニア歴8年の太郎くん

イラストACにて公開の「しのみ」さんのイラストを使用しています。
https://www.ac-illust.com/main/profile.php?id=uCKphAW2&area=1

極大値と極小値の特定する(Python版)

太郎くん

今回は、Python(Numpy)で極大値と極小値の特定するコードだね。

フクさん

早速コードを見せよう。

import numpy as np
import matplotlib.pyplot as plt

N=512;
x=np.linspace(-np.pi,np.pi,N)
fx=np.sin(x)+np.sin(3*x)+np.sin(5*x)
ft=fx;
max_fx=np.max(fx)
max_ft=np.max(ft)
plt.plot(fx)

# 極大値、極小値特定
maxima = np.zeros(N)
minima = np.zeros(N)
fxTmp=fx[0]
mode = 0
for i in range(1,N):
    if mode == 0:
        if fx[i]>fxTmp:
            minima[i]=ft[i]*(max_fx/max_ft)
            mode = 1
        fxTmp=fx[i]
    if mode == 1:
        if fx[i]<fxTmp:
            maxima[i]=ft[i]*(max_fx/max_ft)
            mode = 0
        fxTmp=fx[i];

maxima[maxima==0]=np.nan # 極大値だけを残す
minima[minima==0]=np.nan # 極小値だけを残す
plt.plot(maxima,'ro')  # 極大値をplot
plt.plot(minima,'bo')  # 極大値をplot

plt.show()