r - 使用循环进行 R 编程

Question

我有以下 3x3 方阵a和一个向量x：

a = matrix(c(1.0, 0.1, 0.5, 
             0.1, 1.0, 0.9,
             0.5, 0.9, 1.0), nrow=3, ncol=3)

x = c(0.1,0.2,0.3)

我想基本上使用矩阵和向量的上三角形来遵循公式

y = (a12 - (x1/x2)^2 + (a13 - x1/x3)^2 + (a23 - x2/x3)^2

我在 R 中为上述公式使用了以下 for 循环，我收到以下错误：

Error in a[i, j] : subscript out of bounds

如果我做错了什么，请告诉我，我将不胜感激。

## initialize y
y = 0 

for (i in 1 : nrow(a)-1){
  for (j in i+1 : nrow(a)){
    y = y + ((a[i,j]) - (x[i]/x[j])^2
  }
}

Answer 1

您可以使用以下方法避免循环：

b <- a - outer(x,x,`/`)
sum(b[upper.tri(b)]^2)

结果

[1] 0.2422222

Answer 2

您缺少一些括号，否则n:m将定义一些意想不到的序列，R 看起来像任何其他向量并且可以与其他数学交互。

原始代码，公式替换为打印索引：

for (i in 1 : nrow(a)-1){
+   for (j in i+1 : nrow(a)){
+ print(c(i,j))
+ }
+ }
[1] 0 1
[1] 0 2
[1] 0 3
[1] 1 2
[1] 1 3
[1] 1 4
[1] 2 3
[1] 2 4
[1] 2 5

那些0s 和5s 不在范围内。

但是如果到处都放括号，它的行为是：

for (i in (1 : (nrow(a)-1))){
+   for (j in ((i+1) : nrow(a))){
+ print(c(i,j))
+ }
+ }
[1] 1 2
[1] 1 3
[1] 2 3