Hi all,

I am currently attempting to add a moving average condition to one of the excellent tutorials provided on this platform. As I am still learning, it would be very helpful for me to understand the following. I would like to improve the performance for the algo below (provided in this tutorial: Tutorials - Strategy Library - CAPM Alpha Ranking Strategy on Dow 30... by adding on and testing a simple/or exponential moving average that would serve as a condition. The idea is that the algo would only purchase the selected stocks if the DJI moving average is over 200 days for example. If the condition is not met, the algo would then simply do nothing until the condition is met.

Many thanks for any help on this!

import numpy as np

class BetaAlgorithm(QCAlgorithm):

def Initialize(self):
self.SetStartDate(2016, 1, 1) # Set Start Date
self.SetEndDate(2017, 1, 1) # Set End Date
self.SetCash(10000) # Set Strategy Cash

# Dow 30 companies.
for ticker in ['AAPL', 'AXP', 'BA', 'CAT', 'CSCO', 'CVX', 'DD',
'DIS', 'GE', 'GS', 'HD', 'IBM', 'INTC', 'JPM',
'KO', 'MCD', 'MMM', 'MRK', 'MSFT', 'NKE', 'PFE',
'PG', 'TRV', 'UNH', 'UTX', 'V', 'VZ', 'WMT', 'XOM'] ]

# Benchmark
self.benchmark = Symbol.Create('SPY', SecurityType.Equity, Market.USA)

# Set number days to trace back
self.lookback = 21

# Schedule Event: trigger the event at the begining of each month.
self.Schedule.On(self.DateRules.MonthStart(self.symbols),
self.TimeRules.AfterMarketOpen(self.symbols),
self.Rebalance)

def Rebalance(self):

# Fetch the historical data to perform the linear regression
history = self.History(
self.symbols + [self.benchmark],
self.lookback,
Resolution.Daily).close.unstack(level=0)

symbols = self.SelectSymbols(history)

# Liquidate positions that are not held by selected symbols
for holdings in self.Portfolio.Values:
symbol = holdings.Symbol
if symbol not in symbols and holdings.Invested:
self.Liquidate(symbol)

# Invest 100% in the selected symbols
for symbol in symbols:
self.SetHoldings(symbol, 1)

def SelectSymbols(self, history):
'''Select symbols with the highest intercept/alpha to the benchmark
'''
alphas = dict()

# Get the benchmark returns
benchmark = history[self.benchmark].pct_change().dropna()

# Conducts linear regression for each symbol and save the intercept/alpha
for symbol in self.symbols:

# Get the security returns
returns = history[symbol].pct_change().dropna()
returns = np.vstack([returns, np.ones(len(returns))]).T

# Simple linear regression function in Numpy
result = np.linalg.lstsq(returns, benchmark)
alphas[symbol] = result

# Select symbols with the highest intercept/alpha to the benchmark
selected = sorted(alphas.items(), key=lambda x: x, reverse=True)[:2]
return [x for x in selected]