将线性趋势拟合为一组数据非常简单。但是,我怎样才能将多条趋势线符合一个时间序列呢?我将上下趋势定义为高于或低于指数移动平均线的价格。当价格高于EMA时,我需要拟合一个积极的趋势,当趋势变为负向时,会出现一个新的负面趋势线等等。在我的代码下面,我的熊猫数据框中的market_data['Signal']
告诉我趋势是在+1还是在-1。如何计算和绘制时间序列的多个线性趋势?
我猜我需要某种形式的循环,但我不能工作了逻辑...
import pandas as pd
import pandas_datareader.data as web
import datetime as dt
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.patches as mpatches
import matplotlib.dates as mdates
#Colecting data
market = '^DJI'
end = dt.datetime(2016, 12, 31)
start = dt.date(end.year-10, end.month, end.day)
market_data = web.DataReader(market, 'yahoo', start, end)
#Calculating EMA and difference
market_data['ema'] = market_data['Close'].ewm(200).mean()
market_data['diff_pc'] = (market_data['Close']/market_data['ema']) - 1
#Defining bull/bear signal
TH = 0
market_data['Signal'] = np.where(market_data['diff_pc'] > TH, 1, 0)
market_data['Signal'] = np.where(market_data['diff_pc'] < -TH, -1, market_data['Signal'])
为了适应趋势线我婉使用numpy的polyfit
x = np.array(mdates.date2num(market_data.index.to_pydatetime()))
fit = np.polyfit(x, market_data['Close'], 1)
理想情况下,我想只绘制信号持续超过n个周期的趋势。
结果应该是这个样子:
我不知道我是否已经明白完全...因此,您希望为数据段创建多个线性拟合,每个线段都由'market_data ['Signal']'中的+1或-1定界,这是正确的吗? – jdehesa
是的,这是正确的。理想情况下,只有当我连续超过n +1 ot -1 .. – cJc