매트릭스 리턴 벡터의 논리적 인덱싱 - 초기화 필요?

Question

Matlab에서 행렬에 대한 논리 인덱스를 만들 때 상대 위치를 유지하기 위해 NaN (또는?) 행렬을 초기화해야하는 것처럼 보입니다. 예를 들어, 다음 예제를 고려하십시오

위의 코드에 명시된 바와 같이

clear all; close all; clc; 

Data = [1 2 3 4; 3 40 5 6; 5 6 7 8]; 
ind = logical([0 0 0 0;0 1 1 0;0 1 0 0]); 

tic 
x1 = NaN * ones(3,4); 
y1 = NaN * ones(1,4); y2 = y1; 
x1(ind) = Data(ind); % writing to pre-allocated NaN matrix (necessary?) 
y1(1:4) = nanmean(x1,1); % take NaN-mean - good result 
toc 

% Can we get rid of x1? The following obviously does not work: 
y2(1:4) = nanmean(Data(ind),1); 

, Data(ind)은 원래 크기를 유지하지, 벡터를 반환합니다. 지금까지 가지고있는 유일한 해결책은 올바른 크기로 초기화 된 x1을 사용하는 것입니다. 그래서 : 전체 (x1) NaN 행렬이 필요합니까? 이것이 어떻게 피할 수 있는지에 대한 아이디어가 있습니까? 이 코드의 속도를 높이려고 노력하면서 for 루프를 찾고 있지 않습니다.

Answer 1

accumarray을 사용하면 여기에서 구할 수 있지만 출력물을 미리 할당해야합니다. 그래도 출력물을 사용하면 좋겠지 만 처음부터 필요하지는 않습니다. . 문제의 경우 여기에 두 개의 appproach를 사용할 수 있습니다.

접근 # 1

[~,c] = find(ind); %// get column indices 
out = NaN(1,size(Data,2)); %// pre-allocate for output 

calc_result = accumarray(c,Data(ind),[], @mean) %// get mean calculated results 
array1 = 1:max(c) 
vind = ismember(array1,c) %// valid indices 
out(array1(vind)) = calc_result(vind) 

접근 # 2

[~,c] = find(ind); %// get column indices OR try c = ceil(find(ind)/size(ind,1)) 
vind = c(diff([0 ; c])~=0); %// valid indices 

out(1,1:size(Data,2)) = NaN; %// pre-allocate for output 
calc_result = accumarray(c,Data(ind),[], @mean); %// get mean calculated results 
out(vind) = calc_result(vind); 

---

접근 번호와 신뢰할 수있는 출력 y1와 비교

전체 코드 1 -

clear all; close all; clc; 

% Initialization 
x1 = NaN * ones(3,4); 
y1 = NaN * ones(1,4); 
y2 = y1; 

Data = [ 
    1 2 3 4; 
    3 4 5 6; 
    5 6 7 8] 

% Indices of data to be averaged over rows: 
ind = logical([0 0 0 0;0 1 0 0;0 1 0 1]) 

x1(ind) = Data(ind); % writing to pre-allocated NaN matrix (necessary?) 
y1(1:4) = nanmean(x1,1) % take NaN-mean - good result 

[~,c] = find(ind); %// get column indices 
out = NaN(1,size(Data,2)); %// pre-allocate for output 

calc_result = accumarray(c,Data(ind),[], @mean); 
array1 = 1:max(c); 
vind = ismember(array1,c); 
out(array1(vind)) = calc_result(vind) 

출력 -

우리가 변경할 수

Data = 
    1  2  3  4 
    3  4  5  6 
    5  6  7  8 
ind = 
    0  0  0  0 
    0  1  0  0 
    0  1  0  1 
y1 = 
    NaN  5 NaN  8 
out = 
    NaN  5 NaN  8 

Answer 2

답 : 예, 전체 매트릭스의 경우 정확한 매트릭스 크기와 모양을 얻으려면 사전 초기화가 필요합니다.

메모리 낭비가 염려되는 경우 sparse 명령을 사용하여 스파 스 매트릭스를 생성 할 수 있습니다. 이 경우 논리 인덱스를 별도의 행 및 열 인덱스로 분해해야합니다.