는 2 차원 NumPy와 배열

Question

에서 임의로 선택하는 방법은, ground_truth을 목록의 목록 (평가의도는 정말 매트릭스)를 가지고있다. 나는 비 제로 항목의 20 %를 만들고 싶어 = 0. 나의 초기 접근 방식은 다음과 같습니다

ground_truth = [[0,99,98],[0,84,97], [55,0,0]] 
ground_truth = np.array(ground_truth) 
np.random.choice(ground_truth) 

그러나이

ValueError: a must be 1-dimensional 

그래서 내 솔루션은 1D로 내 매트릭스를 평평하게하는 오류를 제공합니다 배열을 선택하고 0이 아닌 항목의 20 %를 무작위로 선택하십시오.

random_digits = np.random.choice(ground_truth.flatten()[ground_truth.flatten() > 0], 
           int(round(len(ground_truth.flatten()) * .2))) 

in: random_digits 
out: array([99, 97]) 

이제이 항목을 0으로 설정하고 변경 사항을 원본 행렬에 반영하고 싶습니다. 내가 어떻게 할 수 있니?

Answer 1

total_non_zeros = np.count_nonzero(ground_truth) 

# sample 1d index 
idx = np.random.choice(total_non_zeros, int(total_non_zeros * 0.2)) 

# subset non zero indices and set the value at corresponding indices to zero 
ground_truth[tuple(map(lambda x: x[idx], np.where(ground_truth)))] = 0 

ground_truth 
#array([[ 0, 99, 98], 
#  [ 0, 84, 0], 
#  [55, 0, 0]])