Question

我有代码将矩阵的上三角变成向量，并将该向量的值以及它们从矩阵的原始坐标存储到数据帧中。如果向量中的元素为零，如何跳过for循环？

我尝试了else语句和其他尝试。

v <- matrix(sample(0:1, 10, replace = TRUE),9,9)
t <- v[upper.tri(v,diag=T)]
tful <- t[t!=0]
df <- data.frame(FP1=rep(0,length(t)),FP2=rep(0,length(t)),tanimoto=rep(0,length(t)))
for (i in 1:length(t)){
  if (t[i]==0) next
  else {
      col_num <- floor(sqrt(2*i-7/4)+.5)
      row_num <- i-(.5*col_num^2-.5*col_num+1)+1
      df$FP1[i] <- row_num
      df$FP2[i] <- col_num
      df$tanimoto[i] <- v[row_num,col_num]
    }
}

我不希望数据框中的任何零，并且循环跳过这些值。我了解数据框的行数必须较小，但是我以此为例。

Answer 1

您的next工作正常，可以跳过循环的当前迭代。

由于df的所有值都已被df初始化为0，因此最终结果仍然为0。跳过迭代时，它们不会更改，因此它们保持为0。如果将初始化更改为NA值，您会看到没有添加0。

df <- data.frame(FP1=rep(NA,length(t)),FP2=rep(NA,length(t)),tanimoto=rep(NA,length(t)))
for (i in 1:length(t)){
  if (t[i]==0) next
  else {
      col_num <- floor(sqrt(2*i-7/4)+.5)
      row_num <- i-(.5*col_num^2-.5*col_num+1)+1
      df$FP1[i] <- row_num
      df$FP2[i] <- col_num
      df$tanimoto[i] <- v[row_num,col_num]
    }
}
df
#    FP1 FP2 tanimoto
# 1    1   1        1
# 2    1   2        1
# 3    2   2        1
# 4    1   3        1
# 5    2   3        1
# 6    3   3        1
# 7   NA  NA       NA
# 8    2   4        1
# 9    3   4        1
# 10   4   4        1
# 11  NA  NA       NA
# ...

一个简单的修改就是作为最后一步过滤数据框：df = df[df$tanimoto != 0, ]，或者如果您切换到NA，df = na.omit(df)。

我们还可以创建一个非循环解决方案：

v1 = v != 0 
df2 = data.frame(FP1 = row(v)[v1], FP2 = col(v)[v1], tanimoto = v[v1])
df2 = subset(df2, FP1 <= FP2)
df2
#    FP1 FP2 tanimoto
# 1    1   1        1
# 7    1   2        1
# 8    2   2        1
# 13   1   3        1
# 14   2   3        1
# 15   3   3        1
# 20   2   4        1
# 21   3   4        1
# 22   4   4        1
# 27   3   5        1
# 28   4   5        1
# 29   5   5        1
# 33   1   6        1
# 34   4   6        1
# 35   5   6        1
# ...

如果满足条件，如何在for循环中跳过迭代

1 个答案: