विकर्ण पर मूल्य कैसे असाइन करें?

Question

मान लीजिए मैं एक NxN मैट्रिक्स A, एक सूचकांक वेक्टर संख्या 1 के एक सबसेट से मिलकर वी है: लागू नहीं है, और एक मूल्य के कश्मीर, और मैं ऐसा करना चाहते हैं:

for i = V 
    A(i,i) = K 
end 

वहाँ ऐसा करने के लिए एक रास्ता है यह एक कथन w/vectorization में?

उदा। ए (कुछ) = के

कथन A(V,V) = K काम नहीं करेगा, यह ऑफ-विकर्ण तत्वों को निर्दिष्ट करता है, और यह वही नहीं है जो मैं चाहता हूं। उदाहरण के लिए:

>> A = zeros(5); 
>> V = [1 3 4]; 
>> A(V,V) = 1 

A = 

1  0  1  1  0 
0  0  0  0  0 
1  0  1  1  0 
1  0  1  1  0 
0  0  0  0  0 

Answer 1

मैं आमतौर पर उस के लिए EYE का उपयोग करें:

A = magic(4) 
A(logical(eye(size(A)))) = 99 

A = 
    99  2  3 13 
    5 99 10  8 
    9  7 99 12 
    4 14 15 99 

वैकल्पिक रूप से, आप बस रैखिक सूचकांकों की सूची, एक विकर्ण तत्व से के बाद से अगले करने के लिए बना सकते हैं, यह nRows+1 कदम उठा लेता:

[nRows,nCols] = size(A); 
A(1:(nRows+1):nRows*nCols) = 101 
A = 
    101  2  3 13 
    5 101 10  8 
    9  7 101 12 
    4 14 15 101 

आप केवल विकर्ण तत्वों के एक सबसेट उपयोग करना चाहते हैं, तो आप विकर्ण सूचकांक की एक सूची बनाने की जरूरत:

subsetIdx = [1 3]; 
diagonalIdx = (subsetIdx-1) * (nRows + 1) + 1; 
A(diagonalIdx) = 203 
A = 
    203  2  3 13 
    5 101 10  8 
    9  7 203 12 
    4 14 15 101 

वैकल्पिक रूप से, आप एक तार्किक सूचकांक सरणी diag का उपयोग कर बना सकते हैं (वर्ग सरणियों के लिए ही काम करता है)

diagonalIdx = false(nRows,1); 
diagonalIdx(subsetIdx) = true; 
A(diag(diagonalIdx)) = -1 
A = 
    -1  2  3 13 
    5 101 10  8 
    9  7 -1 12 
    4 14 15 101 

Answer 2

>> tt = zeros(5,5) 
tt = 
    0  0  0  0  0 
    0  0  0  0  0 
    0  0  0  0  0 
    0  0  0  0  0 
    0  0  0  0  0 
>> tt(1:6:end) = 3 
tt = 
    3  0  0  0  0 
    0  3  0  0  0 
    0  0  3  0  0 
    0  0  0  3  0 
    0  0  0  0  3 

और अधिक सामान्य:

>> V=[1 2 5]; N=5; 
>> tt = zeros(N,N); 
>> tt((N+1)*(V-1)+1) = 3 
tt = 
    3  0  0  0  0 
    0  3  0  0  0 
    0  0  0  0  0 
    0  0  0  0  0 
    0  0  0  0  3 

इस पर आधारित है तथ्य यह है कि मैट्रिस को एक-आयामी सरणी (वैक्टर) के रूप में एक्सेस किया जा सकता है, जहां 2 सूचकांक (एम, एन) को रैखिक मैपिंग एम * एन + एन द्वारा प्रतिस्थापित किया जाता है।

Answer 3

A = zeros(7,6); 
V = [1 3 5]; 

[n m] = size(A); 
diagIdx = 1:n+1:n*m; 
A(diagIdx(V)) = 1 

A = 
    1  0  0  0  0  0 
    0  0  0  0  0  0 
    0  0  1  0  0  0 
    0  0  0  0  0  0 
    0  0  0  0  1  0 
    0  0  0  0  0  0 
    0  0  0  0  0  0 

Answer 4

मान लीजिए के मूल्य है। आदेश

A=A-diag(K-diag(A)) 

थोड़ा तेजी से

>> A=randn(10000,10000); 

>> tic;A(logical(eye(size(A))))=12;toc 

बीता हुआ समय .५,१७,५७५ सेकंड है हो सकता है।

>> tic;A=A+diag((99-diag(A)));toc 

विलुप्त समय 0.353408 सेकंड है।

लेकिन यह अधिक स्मृति का उपभोग करता है।

Answer 5

मैं sub2ind का उपयोग करें और के रूप में दोनों x और y मापदंडों विकर्ण सूचकांक दे देते हैं:

A = zeros(4) 
V=[2 4] 

idx = sub2ind(size(A), V,V) 
% idx = [6, 16] 

A(idx) = 1 

% A = 
% 0  0  0  0 
% 0  1  0  0 
% 0  0  0  0 
% 0  0  0  1