def GetKKLine(data,period): kk = [] yy = [] log_y = [] index = 0 datalen = len(data) while index<datalen: if index < datalen - period: data_line = data[index:index+period] data_line1 = np.array(data_line) data_line1 = np.log(data_line1) res = myLine(data_line1); yy.append(res['y']) detal = (math.atan(res['k'])/3.1415926)*180 kk.append(detal) log_y.append(data_line1[0]) else : yy.append(np.nan) kk.append(np.nan) log_y.append(math.log(data[index])) index+=1 return kk,yy,log_y
def GetDiffLine(data,period): diff = [] index = 0 datalen = len(data) while index<datalen: if index < datalen - period: data_line = data[index:index+period] data_line1 = np.array(data_line) data_line1 = np.log(data_line1) res = myLine(data_line1); k = res['k']; b = res['b']; diff.append(period*k/b ) else : diff.append(np.nan) index+=1 return diff