Evaluieren arrayfun nur bei Nicht-Null-Einträge zu unterschiedlichen Ergebnissen

Question

I zwei Vektoren x gegeben werde, y und i schaffen die Distanzmatrix

B=pdist2(x,y); 

Jetzt bewerten i eine bestimmte Funktion f (x),

s=5; 
if s-x > 0 
    y=(1-x/s)^4*(1+4*x/s)/20; 
else 
    y=0; 
end 

wo ist s etwas parameter. Durch Vordefinieren von s und Verwenden von

ist die Matrix A positiv definierbar wie gewünscht.

Da die Matrix spärlich ist, möchte ich f (x) nur bei den Werten auswerten, die ungleich Null zurückgegeben werden. Mein Versuch war

B=pdist2(x,y); 
B(B>s)=0; 
B=B/s; 
indexB=find(B); 
Atmp=arrayfun(@f,B(indexB)); 
A(indexB)=Atmp; 

Aber jetzt A ist nicht positiv definit und hat weniger Einträge ungleich Null als zuvor. Kann mir jemand erklären, was ich mache?

Danke allen, ich habe den Fehler gefunden. Da waren einige Null Einträge auf der Diagonale in B, die ich separat berechnen musste.

Answer 1

Sie sollten die Matrix A mit der gleichen Größe wie B preallocate, sonst MATLAB nicht die richtige Größe weiß für A:

A=zeros(size(B)); 

Beachten Sie, dass find kehrt linear indices, so A einen Vektor anstelle einer wurde Matrix.

Answer 2

Ich frage mich, warum Sie versuchen, Index B, und wenden Sie dann eine Funktion auf jedes einzelne Element. Stattdessen lassen Sie uns vectorize Code

% Same setup 
B = pdist2(x,y); s = 5; 
B(B >= s) = 0; B = B/s; 
% Don't use arrayfun, instead use vectorised code 
% The previous line already dealt with the 'else' case from f(x) 
B = ((1-B/s).^4).*(1+4*B/s)/20; 

Beachten Sie, dass jedes Mal, wenn wir zwischen zwei nicht-Skalare arbeiten wir elementweise Operationen .^ oder .* verwenden. Die anderen Divisionen und Multiplikationen sind alle mit Skalaren, aber wir könnten sicher sein und uns angewöhnen, sie alle zu benutzen!

B = ((1-B./s).^4).*(1+4.*B./s)./20; 

---

Ein Hinweis auf MATLAB Terminologie: Seien Sie vorsichtig sagen, dass Sie eine "Sparse Matrix" haben hier, das ist ein special variable type in MATLAB. Sie haben eine Standardmatrix, die mathematisch spärlich ist! Das Arbeiten auf einer spärlichen Matrix könnte etwas anders sein.

Answer 3

Statt arrayfun können Sie spfun verwenden, um eine Funktion zu Null verschiedene Sparse Matrix Elemente anwenden:

A = spfun(@f, B); 

Obwohl Vektorisierung effizienter sein kann.

• spfun arbeitet auch mit dichten Matrizen sowie spärlich diejenigen, aber Ausgang es eine spärliche Matrix

Dank @Wolfie einige Tests zur Verfügung gestellt und für ihre Klarstellung.