绘制连接时间序列中大多数数据点的线

Question

我想知道如何绘制最佳拟合线或连接最多点的线；一条线表示局部最小值，另一条线表示局部最大值。

就像这张图片一样：

以下是查找局部最小值和最大值点的代码

import numpy as np
import pandas as pd  
import matplotlib.pyplot as plt
from pandas_datareader import data as pdr
import fix_yahoo_finance as yf
yf.pdr_override() 

startdate = dt.date(2016, 1, 1)
#enddate = dt.date(2018, 10, 22)
today = dt.date.today()
index_ticker = "^GSPC"
index_df = pdr.get_data_yahoo(index_ticker, start=startdate, end=today)

from scipy.signal import argrelextrema
n=50 # number of points to be checked before and after 

# Find local peaks/bottoms
index_df['min'] = index_df.iloc[argrelextrema(index_df['Adj Close'].values, np.less_equal, order=n)[0]]['Adj Close']
index_df['max'] = index_df.iloc[argrelextrema(index_df['Adj Close'].values, np.greater_equal, order=n)[0]]['Adj Close']

# Plot results
plt.scatter(index_df['Adj Close'].index, index_df['min'], c='r')
plt.scatter(index_df['Adj Close'].index, index_df['max'], c='g')
plt.plot(index_df['Adj Close'].index, index_df['Adj Close'])
plt.show()

我遇到了这个教程，https://prappleizer.github.io/Tutorials/Plotting/Plotting.html，它绘制了一条具有不确定性阴影区域的最佳拟合线。

这就是我正在寻找的，但我不知道如何应用于时间序列以获得时间序列图片中所需的输出。

如果有人可以提供建议，我们将不胜感激。谢谢！

Answer 1

现在您已经有了局部最小值和最大值，请在每组点上绘制线性回归。使用您最喜欢的 Python 统计包来执行此操作。