2dのnumpyの配列からランダムに選択する方法

Question

私はリストのリストを持っています（実際には格付けの行列です）、ground_truthです。しかし、これはエラーに

ValueError: a must be 1-dimensional 

を与えるだから私のソリューションは、1Dに私の行列を平坦化することである

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

：私は= 0.私の最初のアプローチは、非ゼロ項目の20％を作りたいです次に、非ゼロ項目の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]])