Wstaw numer powyżej przekątnej macierzy R

Question

Mam wektor takiego w R:

vec1 <- c(14000,12000,8000) 

Próbuję utworzyć macierz gdzie 14000 to mój główny przekątna, 1200 to jeden powyżej przekątnej, 8000 dwa powyżej przekątnej.

Jestem zaznajomiony z robieniem tego w Pythonie/numpy, ale nie mogę go rozgryźć w R (lub przynajmniej skuteczny sposób to zrobić). Idealnie wyjście będzie wyglądać następująco:

14000 12000 8000 
    0 14000 12000 
    0  0 14000 

Answer 1

Spróbuj

m1 <- t(matrix(vec1, nrow=5, ncol=3))[,1:3] 
m1[lower.tri(m1)] <- 0 
m1 
#  [,1] [,2] [,3] 
#[1,] 14000 12000 8000 
#[2,]  0 14000 12000 
#[3,]  0  0 14000 

, lub toeplitz

toeplitz(vec1)*upper.tri(diag(seq_along(vec1)), diag=TRUE) 
#  [,1] [,2] [,3] 
#[1,] 14000 12000 8000 
#[2,]  0 14000 12000 
#[3,]  0  0 14000 

lub zmodyfikowanym sugerowane przez @David Arenburg

m <- toeplitz(vec1) 
m[lower.tri(m)] <- 0 

Answer 2

W tym prostym przypadku można przypisać ukośne i górne trójkątne porcje oddzielnie tak:

m <- matrix(0, nrow=3, ncol=3) 

diag(m) <- 14000 

m[upper.tri(m)] <- c(12000, 8000, 12000) 

Jednakże, jak David Arenburg podkreślić, że jest to bardziej podejście ręczne, a tym samym nie skaluje się dobrze. Aby uzyskać bardziej zautomatyzowane i skalowalne podejście, polecam rozwiązanie firmy Akrun, używając toeplitz() i lower.tri().

Będę tu trzymał tę odpowiedź ze względu na jej kompletność, ale uważam, że akrun jest lepszym ogólnym rozwiązaniem.