在Matlab中選擇列表中隨機非零元素的索引

Question

是否有任何準備好的命令從輸入中獲取預期的輸出？

輸入

>>> a=[1 0 3 0 5 6 7 8 0 10 0 0]; selectNonZero(a) 

預期輸出

1 or 3 or 5 or 6 or 7 or 8 or 10 

試驗

>> b=a(a~=0); pi=randi([1, length(b)]); b(pi) % The original index of b(pi)? 

>> fix=[0 1 2 2 2 2 2]; pi+fix(pi)    % Fix changed index, cum command? 

Answer 1

路易斯的回答的一個變體是內置nnz功能的使用方法：

idx = find(a); 
rand_idx = idx(randi(nnz(a))); 

如果你已經安裝了統計工具箱，可以減少到一個班輪這種使用randsample：

rand_idx = randsample(find(a), 1); 

注意：如果您想選擇隨機值（而不是其索引），請將find(a)替換爲nonzeros(a)（這是生成隨機索引然後執行a(rand_idx)）。

Answer 2

你可以這樣來做。這與您的方法類似，但使用find來了解非零值的索引。

jj = find(a~=0); % indices of nonzero values of a 
ind = jj(randi(length(jj))); % randomly pick one of those indices 
val = a(ind); % corresponding value of a 

你想要的結果sel（選擇值）和ind（其內a指數）。