는 어떻게 다음 줄을 사용

Question

배열의 목록에 배열의 현재 위치에서 항목을 삭제하기 .
이것은 각 배열의 0 번째 인덱스 (첫 번째 배열 제외)가 0 인 배열의 목록을 제공합니다. 자리 표시 자 0이없는 배열 목록을 얻으려면이 0을 제자리에서 삭제하려면 어떻게합니까? 내가 반복하지 않기 때문에 각 배열에 삭제 된 0을 새 배열 목록에 추가하기 때문에 내부에서 말합니다.

당신은 0을 제거하기 위해 몇 가지 사전 처리를 사용할 수 있습니다

Answer 1

, 여기에 예입니다

import numpy as np 
a = np.random.randint(0, 5, 100) 
idx = np.where(a == 0)[0] 

np.split(a, idx) 

출력 :

[array([4, 3]), 
array([0, 3, 1, 1]), 
array([0, 4]), 
array([0]), 
array([0, 1, 2, 4]), 
array([0, 4, 2, 2, 3, 1]), 
array([0, 2, 4, 3]), 
array([0, 2, 2, 4]), 
array([0, 3, 2, 1, 2, 4, 1, 2]), 
array([0]), 
array([0]), 
array([0, 1, 3]), 
array([0]), 
array([0, 1, 3, 2, 4]), 
array([0, 3, 3]), 
array([0, 3]), 
array([0, 3, 4, 1, 2, 3, 3, 4, 3, 4, 2, 1]), 
array([0]), 
array([0, 3, 2]), 
array([0]), 
array([0, 4, 1, 4, 2, 3, 1]), 
array([0, 4]), 
array([0]), 
array([0]), 
array([0, 3, 4, 2, 3]), 
array([0, 4, 4, 3, 3, 3, 1]), 
array([0, 2, 4, 1]), 
array([0, 4, 2, 2]), 
array([0])] 

제거 a의 모든 제로, 새로운 인덱스 계산 :

idx2 = idx - np.arange(len(idx)) 
b = np.delete(a, idx) 
np.split(b, np.unique(idx2)) 

을

다음은 출력입니다.

[array([4, 3]), 
array([3, 1, 1]), 
array([4]), 
array([1, 2, 4]), 
array([4, 2, 2, 3, 1]), 
array([2, 4, 3]), 
array([2, 2, 4]), 
array([3, 2, 1, 2, 4, 1, 2]), 
array([1, 3]), 
array([1, 3, 2, 4]), 
array([3, 3]), 
array([3]), 
array([3, 4, 1, 2, 3, 3, 4, 3, 4, 2, 1]), 
array([3, 2]), 
array([4, 1, 4, 2, 3, 1]), 
array([4]), 
array([3, 4, 2, 3]), 
array([4, 4, 3, 3, 3, 1]), 
array([2, 4, 1]), 
array([4, 2, 2]), 
array([], dtype=float64)]