输入tjpd  输出  a,b
tjpd    a   b
0   0   1
0   0   2
0   0   3
0   0   4
1   1   5
0   0   6
0   0   7
0   0   8
0   0   9
1   2   10
0   0   11
0   0   12
0   0   13
1   3   14
需要在vnpy框架下写,请帮检查一下这个函数的一些问题:
1、tjpd是个条件数组,这样定义对吗?
2、a、b是否要定义初始值呢?
def ceshi(
    self,
    tjpd: bool,
    array: bool = False
) -> Union[
    Tuple[np.ndarray, np.ndarray],
    Tuple[int, int]
]:
    """
    ceshi
    """
    i=0
    j=0
    while(a[-i]<=3):
        b[-i-1]=b[-i]+1
        if tjpd[-i]:
            a[-i]=j+1
            j=j+1
        else :
            a[-i]=0
        i=i+1
    if array:
        return a, b
    return a[-1], b[-1]
                        
                        
                        
                        
                    