2013-01-14 38 views
8

我需要访问并分配for循环中的m * n矩阵的单个槽。到目前为止的代码:将值分配给矩阵的对角条目

rowCount <- 9 
similMatrix = matrix(nrow = rowCount - 1, ncol = rowCount) 
show(similMatrix) 
for(i in (rowCount - 1)){ 
    for (j in rowCount) 
    if (i == j){ 
     similMatrix[i == j] <- 0; 
    } 
} 
show(similMatrix) 

所以,如果我= j的矩阵中的NA值需要与0

+0

你是不是获得期望的输出? –

回答

10

为目的将“对角线”元素设置为零,你已经得到了答案,但我想知道你是否希望得到更一般的东西。该代码缺乏成功的原因有两方面:你的索引的构建有缺陷,并且索引是错误的。这将成功:

for(i in 1:(rowCount - 1)){ # need an expression that retruns a sequence 
    for (j in 1:rowCount)  # ditto 
    if (i == j){ 
     similMatrix[i,j] <- 0; # need to index the matrix with two element if using i,j 
    } 
} 
#---------- 
> show(similMatrix) 
    [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] 
[1,] 0 NA NA NA NA NA NA NA NA 
[2,] NA 0 NA NA NA NA NA NA NA 
[3,] NA NA 0 NA NA NA NA NA NA 
[4,] NA NA NA 0 NA NA NA NA NA 
[5,] NA NA NA NA 0 NA NA NA NA 
[6,] NA NA NA NA NA 0 NA NA NA 
[7,] NA NA NA NA NA NA 0 NA NA 
[8,] NA NA NA NA NA NA NA 0 NA 

但诉诸于R键循环通常被认为是最后的手段有做同样的“循环”操作的更加紧凑的方式,它概括(有时错误的原因。)比设定对角线更广泛。

similMatrix[ row(similMatrix) == col(similMatrix) ] <- 0 
> similMatrix 
    [,1] [,2] [,3] [,4] [,5] [,6] [,7] [,8] [,9] 
[1,] 0 NA NA NA NA NA NA NA NA 
[2,] NA 0 NA NA NA NA NA NA NA 
[3,] NA NA 0 NA NA NA NA NA NA 
[4,] NA NA NA 0 NA NA NA NA NA 
[5,] NA NA NA NA 0 NA NA NA NA 
[6,] NA NA NA NA NA 0 NA NA NA 
[7,] NA NA NA NA NA NA 0 NA NA 
[8,] NA NA NA NA NA NA NA 0 NA 

如果你想设置的次对角为零,你可以只使用:

similMatrix[ row(similMatrix)-1 == col(similMatrix) ] <- 0 

可避免产生多余的行和使用列矩阵这样的:

mind <- min(dim(similMatrix)) 
# avoid going outside dimensions if not symmetric 
similMatrix[ cbind(seq(maxd),seq(maxd)) <- 0 
+0

这是完美的,非常感谢你! –

+1

我想'行(mat)'和'col(mat)'会生成两个矩阵,这是不是意味着我们在这里内存消耗增加了三倍 ?如果'垫'发生 是大,我想这可能会导致问题。 – qed

+0

非常正确。我没有看到任何警告,这些是大矩阵。我提供的“紧凑性”是表达式,但正如你所指出的那样,这可能不是紧凑的内存占用。我们可以想象使用'similMatrix [cbind(1:(rowCount - 1),1:(rowCount - 1)] < - 0',我猜测它对内存和速度都会表现出色。认为这可能会有更小的占地面积,因为这种“垫[diag(mat)] <-0”方法。 –

27

替换你想要的功能diag<-

m <- matrix(1:12, nrow=3) 
m 
    [,1] [,2] [,3] [,4] 
[1,] 1 4 7 10 
[2,] 2 5 8 11 
[3,] 3 6 9 12 

diag(m) <- 0 
m 
    [,1] [,2] [,3] [,4] 
[1,] 0 4 7 10 
[2,] 2 0 8 11 
[3,] 3 6 0 12 
+0

“删除我的答案 - 并提出其他”按钮在哪里? – mnel

+0

谢谢,它的工作原理,但我现在如何分配一个值让我们说m [1,2]? –