numpy 배열의 논리적 반복

Question

두 개의 같은 크기의 목록이 있다고 가정합니다. 첫 번째 목록에는 0과 1 만 포함되며 두 번째 목록의 초기 값은 고정 된 숫자와 같습니다. 두 번째 배열의 다른 값은 첫 번째 목록의 동일한 색인 값에 따라 다릅니다. 이들 사이의 관계는 첫 번째 목록의 값이 0 인 경우 두 번째 목록의 동일한 색인 값이 이전 색인과 같고 다른 모든 경우는 다른 값과 같습니다. 내 질문을 명확히하기 위해 for 루프의 도움으로 아래 코드를 작성했습니다. for 루프없이 문제를 해결하는 방법은 무엇입니까?

Code 
a = np.array([0, 1, 0, 0, 0, 1, 0, 0, 1]) 
b = np.zeros_like(a) 
b[0] = 5 
for i in range(1, a.size): 
    if a[i] == 0: 
     b[i] = b[i-1] 
    else: 
     b[i] = np.random.randint(5) 

Answer 1

는 여기 벡터화 된 접근 방식 -

offset = int(a[0]!=0) 
N = (np.count_nonzero(a!=0)) - offset # no. of rand num to be generated 
rand_num = np.append(5,np.random.randint(0,5,N)) 
out = rand_num[(a!=0).cumsum() - offset]