breakout.py

for i in range(0,len(local_max)-3):
  (m,c),r,_,_,_= np.polyfit(local_max_idx[i:i+3],local_max[i:i+3],1,full=True)
  if(m<=3 and m>=-3 and (r[0]<20 and r[0]>-20)):
    start=local_max_idx[i+2]
    for k in range(start,start+7):
      if(k<len(prices_close) and prices_close[k]>(k*m+c)):
        plt.figure(figsize=(10,5))
        plt.plot(local_max_idx,m*local_max_idx+c,'m')
        plt.plot(prices_close)
        plt.plot(k,prices_close[k],'bo')
        break