Question

R版本3.3.2

我正在尝试根据该数据框的其他变量的值，使用嵌套的for循环将某些值分配给我的数据框的空变量。然而，输出并不是我所期望的。

这是一个可复制的例子：

id <- c("ID61", "ID61", "ID63", "ID69", "ID69", "ID69", "ID69", "ID69", "ID80", "ID80", "ID80", "ID81", "ID81", "ID81", "ID81")
Round <- c(1, 2, 1, 1, 2, 3, 4, 5, 1, 2, 3, 1, 2, 3, 4)
nrPosRound <- c(2, 0, 2, 15, 8, 4, 4, 0, 3, 1, 1, 0, 0, 0, 0)
Y <- rep(NA, 15)
df <- data.frame(id, Round, nrPosRound, Y)

我的数据框看起来像这样：

> df
     id Round nrPosRound Y
1  ID61     1          2 NA
2  ID61     2          0 NA
3  ID63     1          2 NA
4  ID69     1         15 NA
5  ID69     2          8 NA
6  ID69     3          4 NA
7  ID69     4          4 NA
8  ID69     5          0 NA
9  ID80     1          3 NA
10 ID80     2          1 NA
11 ID80     3          1 NA
12 ID81     1          0 NA
13 ID81     2          0 NA
14 ID81     3          0 NA
15 ID81     4          0 NA

在嵌套的for循环之后，我希望它看起来像这样：

> df
     id Round nrPosRound Y
1  ID61     1          2 FP
2  ID61     2          0 FP
3  ID63     1          2 FP
4  ID69     1         15 FP
5  ID69     2          8 FP
6  ID69     3          4 FP
7  ID69     4          4 FP
8  ID69     5          0 FP
9  ID80     1          3 1
10 ID80     2          1 1
11 ID80     3          1 1
12 ID81     1          0 0
13 ID81     2          0 0
14 ID81     3          0 0
15 ID81     4          0 0

我想要的是分配值＆＃39; 1＆＃39;变量＆＃39; Y＆＃39;如果，对于同一个＆＃39; id，在某个＆＃39;中，有3个或更多个阳性（nrPosRound＆gt; = 3），并且在接下来的回合中至少有1个阳性（nrPosRound＆gt; = 1）。＆＃39; Y＆＃39;会被分配值＆＃39; 0＆＃39;如果，在每个轮次＆＃39;对于相同的＆＃39; id＆＃39; ＆＃39; nrPosRound＆＃39;是＆＃39; 0＆＃39;。＆＃39; Y＆＃39;应该分配＆＃39; FP＆＃39; （误报）如果不满足以前的条件。如果只有1＆＃39; Round＆＃39;为了那个＆＃39; id＆＃39;，＆＃39; Y＆＃39;会有价值＆＃39; 1＆＃39;如果＆＃39; nrPosRound＆＃39;是＆gt; = 3;价值＆＃39; 0＆＃39;如果＆＃39; nrPosRound＆＃39; == 0;价值＆＃39; FP＆＃39;如果＆＃39; nrPosRound＆＃39; ＆lt; = 3.

这是我的代码，嵌套for循环：

for (i in 1:nrow(df)) {

  current_id <- df$id[i]
  id_group <- df[df$id == curr_id, ]

  for (j in 1:nrow(id_group)) {

    current_Round <- id_group$Round[j] 
    remainder_Rounds <- id_group$Round[(j+1):nrow(id_group)]

    current_nrPos <- id_group$nrPosRound[id_group$Round == current_Round]
    remainder_nrPos <- id_group$nrPosRound[id_group$Round %in% remainder_Rounds]

    ifelse(curr_nrPos >= 3 & remainder_nrPos >= 1,
           df$Y[i] <- 1, ifelse(curr_nrPos == 0 & remainder_nrPos == 0,
                                        df$Y[i] <- 0, "FP"))
  }
}

我认为问题与＆＃39; remainder_nrPos＆＃39;有关，因为第二个ifelse并不像我希望的那样工作。我尝试了很多方法，但似乎无法像我预期的那样使它工作。任何帮助表示赞赏！

Answer 1

可以使用dplyr完成此操作。在以下代码中，我首先group_by id。

我创建一个中间变量min_from_last，看看每轮后是否有零。为此，我首先使用arrange(desc(Round))重新排序。之后我使用cummin来获得累积分钟。

然后，我重新排序数据并执行三个ifelse以获得您想要的结果。顺便说一下，你可能不需要第二个ifelse，因为它会被第一个ifelse捕获，但我把它包含在你的问题中。

id <- c("ID61", "ID61", "ID63", "ID69", "ID69", "ID69", "ID69", "ID69", "ID80", "ID80", "ID80", "ID81", "ID81", "ID81", "ID81")
Round <- c(1, 2, 1, 1, 2, 3, 4, 5, 1, 2, 3, 1, 2, 3, 4)
nrPosRound <- c(2, 0, 2, 15, 8, 4, 4, 0, 3, 1, 1, 0, 0, 0, 0)
df1 <- data.frame(id, Round, nrPosRound,stringsAsFactors=FALSE)

library(dplyr)
df1 %>%
group_by(id) %>%
arrange(desc(Round)) %>%
mutate(min_from_last=cummin(nrPosRound)) %>%
arrange(Round)  %>%
mutate(Y= ifelse(max(nrPosRound)>=3 & min_from_last>0 ,"1",
           ifelse(n()==1 & nrPosRound>=3,"1",
           ifelse(max(nrPosRound)==0,"0","FP"))))

      id Round nrPosRound min_from_last     Y
   (chr) (dbl)      (dbl)         (dbl) (chr)
1   ID61     1          2             0    FP
2   ID61     2          0             0    FP
3   ID63     1          2             2    FP
4   ID69     1         15             0    FP
5   ID69     2          8             0    FP
6   ID69     3          4             0    FP
7   ID69     4          4             0    FP
8   ID69     5          0             0    FP
9   ID80     1          3             1     1
10  ID80     2          1             1     1
11  ID80     3          1             1     1
12  ID81     1          0             0     0
13  ID81     2          0             0     0
14  ID81     3          0             0     0
15  ID81     4          0             0     0

Answer 2

这是一个基础R解决方案。

id.vals <- unique(df$id)

for (i in 1:length(id.vals)) {
    group.ind <- df$id == id.vals[i]
    id_group  <- df[group.ind, 'nrPosRound']
    n   <- length(id_group)
    Y   <- rep(NA, n)
    g3  <- any(id_group >= 3)
    a0  <- all(id_group == 0)
    for (j in 1:n) {
        if (g3 & all(id_group[j:n] >= 1)) Y[j] <- 1
        else if (a0) Y[j] <- 0
        else Y[j] <- 'FP'
    }
    df$Y[group.ind] <- Y
}

R - 为嵌套for循环

2 个答案: