【问题标题】:how to store result in list from a for loop如何将结果从for循环存储到列表中
【发布时间】:2015-05-07 19:43:13
【问题描述】:

我是 R 新手,这可能是一个愚蠢的问题,但我不知道如何解决。

我有一个 for 循环,它可能会返回 i*j 非空元素。

我想将所有非空结果存储在一个列表中,但是如果我在循环中使用result[[i]]<-tmp,它最多只能存储i 元素,我如何能够将所有值存储在一个列表中?谢谢

  result<-list()
  for (i in 1:nrow(m)){
    for (j in 1:i){
      if(m[i,j]!=0 && m[j,i]!=0){
        num=min(m[i,j],m[j,i])
        tmp=c(i,j,num)
        result[[i]]<-tmp
      }
    }
  }

sample data
set.seed(123)
m= matrix(sample(0:5, size = 5*5, replace = TRUE), ncol = 5)

想要的

 row col min
[1] 1 1 1
[1] 2 2 3
[1] 3 1 2
[1] 3 2 2
[1] 3 3 4
[1] 4 1 5
[1] 4 2 1
[1] 4 4 1
[1] 5 1 5
[1] 5 2 2
[1] 5 4 5
[1] 5 5 3

根据大卫的回答

pmin(mx[upper.tri(mx, diag = TRUE)], mx[lower.tri(mx, diag = TRUE)])

[1] 1 0 2 5 2 3 5 1 0 1 3 0 1 5 3

返回

      > result
[[1]]
[1] 1 1 2

[[2]]
[1] 2 2 3

[[3]]
[1] 3 3 4

[[4]]
[1] 4 4 2

[[5]]
[1] 5 5 4

【问题讨论】:

  • R 中的 For 循环和不断增长的列表都非常慢。您能否将您的算法重新考虑为应用于矩阵每个元素的函数并使用apply()?
  • 你不是基本在计算pmin(m, t(m))吗?
  • 我不确定您要做什么以及什么不起作用。您也不需要如此大的数据集来重现您的问题。我认为像m = matrix(sample.int(5, size = 5*5, replace = TRUE), ncol = 5) 这样的东西应该做。还可以使用set.seed 并显示您期望从for 循环中得到什么。
  • 谢谢大家,我正在尝试比较沿矩阵对角线对称位置的值并返回最小值...
  • pmin(m[upper.tri(m, diag = TRUE)], m[lower.tri(m, diag = TRUE)]) 怎么样?

标签: r loops


【解决方案1】:

这类似于@DavidArenburg 的回答(从评论转换而来):

idx <- which(upper.tri(m,diag=TRUE),arr.ind=TRUE)
v   <- pmin(m[idx], m[idx[,2:1]])

cbind(idx,min=v)[v>0,]

给了

      row col min
 [1,]   1   1   1
 [2,]   2   2   3
 [3,]   1   3   2
 [4,]   2   3   2
 [5,]   3   3   4
 [6,]   1   4   5
 [7,]   2   4   1
 [8,]   4   4   1
 [9,]   1   5   5
[10,]   2   5   2
[11,]   4   5   5
[12,]   5   5   3

【讨论】:

  • 这是一个非常好的。我刚刚发布了一个类似的答案,但是which 的这种很好的用法要好得多。我打算和tm &lt;- t(m) ; pmin(tm[upper.tri(tm, diag = TRUE)], m[upper.tri(m, diag = TRUE)])
猜你喜欢
  • 2015-08-24
  • 2015-12-14
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
  • 1970-01-01
相关资源
最近更新 更多