输入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]