変化点に点線を引きたい

Question

変化点に点線を引きたいです。
https://qiita.com/deaikei/items/af728ab0d43ca97bc1f6　を参考にして異常値検知を行なっています。

# -*- coding: utf-
import matplotlib.pyplot as plt
import changefinder
import numpy as np
data=np.concatenate([np.random.normal(0.7, 0.05, 300), 
np.random.normal(1.5, 0.05, 300),
np.random.normal(0.6, 0.05, 300),
np.random.normal(1.3, 0.05, 300)])

cf = changefinder.ChangeFinder(r=0.01, order=1, smooth=7)

ret = []
for i in data:
    score = cf.update(i)
    ret.append(score)

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(ret)
ax2 = ax.twinx()
ax2.plot(data,'r')
plt.show()

のようにコードを書きました。
その結果、

のようになると思うのですが、赤い矢印で示した黒い点線のように変化点に点線を引きたいです。

どのようにコードを書けば、それができますか？

Answer 1

changefinderによって得られた変化点のscoreリストを元データとして、scipy.signal.argrelmaxによってその極大値の位置リストを求めます。あとは位置リストをもとに縦線を描画します。
参考：SciPy で離散データのピークを検出

import matplotlib.pyplot as plt
import changefinder
import numpy as np
data=np.concatenate([np.random.normal(0.7, 0.05, 300), 
    np.random.normal(1.5, 0.05, 300),
    np.random.normal(0.6, 0.05, 300),
    np.random.normal(1.3, 0.05, 300)])

cf = changefinder.ChangeFinder(r=0.01, order=1, smooth=7)
ret = []
for d in data:
    score = cf.update(d)
    ret.append(score)
ret = np.array(ret) # 後の処理のためnumpy arrayに

# 極大値のx位置を取得
# order値＝次の極大値の判定範囲。ちょっと恣意的だが200を与える…
from scipy import signal
maxId = signal.argrelmax(ret,order=200)
print(maxId)

fig = plt.figure()
ax = fig.add_subplot(111)
ax.plot(ret)
# 縦線描画
for i in maxId[0]:
    ax.plot([i, i],[0, 100], "black", linestyle='dashed')

ax2 = ax.twinx()
ax2.plot(data,'r')
plt.show()

Comments

[taiyo2017]

わかりやすい回答をありがとうございます。１点質問です。maxId = signal.argrelmax(ret,order=200)　のorderで範囲指定し、その範囲の中から極大を見つけるようにすることはできますか？例えば、５０〜１５０の間で極大を見つけるようにする、ためにsignal.argrelmax(ret,order=[50,150])のように書いて実行する、みたいに。

[can110]

詳細はargrelmaxマニュアル参照されたほうが確実ですが、たぶんできないと思います。
その代わりret自体を50～150の範囲にフィルタリングしたうえでargrelmax呼び出すのが良い筋かと思います。

[taiyo2017]

承知しました。ありがとうございます。