現在、tweetの感情の時系列と株価の時系列データを比較し、感情の変化がどのくらい株価と連動しているのかの分析をやっており、DTWで2つの時系列間の類似度を算出したのですが、図のようにある感情の点(オレンジ)がそれよりも前の時間の株価(青)の点と対応してしまいます。なので感情の点とそれよりも後の時間の株価の点を対応付けるように探索範囲を指定したいのですが、やり方が分からないのでご教授願いたいです。
環境はGoogle colabです
python
1from fastdtw import fastdtw 2import matplotlib.pyplot as plt 3# グラフを横長固定(お好みで) 4from matplotlib.pylab import rcParams 5rcParams['figure.figsize'] = 10, 6 6 7a = tw_df1['finish'].to_list() #株価 8b=tw_df1['PN'].to_list() #感情 9x1=tw_df1['time'] 10 11distance, path = fastdtw(a, b) #DTW距離を求める 12 13plt.plot(a, label="finish") #株価 14plt.plot(b, label="PN") #感情 15 16for a_x, b_x in path: 17 plt.plot([a_x, b_x], [a[a_x], b[b_x]], color='gray', linestyle='dotted', linewidth=1) 18 19plt.legend() 20plt.show() 21 22
あなたの回答
tips
プレビュー