direction_indicators.py 文件源码

def _build_indicator(self, span, **kwds):
        """indicator????????????

        Args:
            span: ??????????
        """
        def get_direction(val1, val2):
            if np.isnan(val1) or np.isnan(val1):
                return np.nan
            elif val1 < val2:
                return const.INDI_DIRECTION_UP  # ???
            elif val1 > val2:
                return const.INDI_DIRECTION_DOWN  # ???
            else:
                return const.INDI_DIRECTION_HR  # ??

        ma = MovingAverageIndicator(stock=self.stock, span=span)
        arr1 = ma.shifted(-1)  # ???????????
        arr2 = ma.data  # ???????
        return np.array([get_direction(a, b) for a, b
                         in zip(arr1, arr2)], dtype=np.float16)