Увеличение счетчика в цикле for в R

Question

Привет всем, у меня есть проблема с счетчиком, который я создал в цикле for. Вот код.

n1<- 0 
n2<- 0 
n<- 20 
f<- rep(NA,20) 
set.seed(10) 
x<- runif(20) 
for(i in 1:20) { 
f[i]<- ifelse(x[i]<0.5, 'a', 'b') 
if((n1-n2)==2) { 
break 
} 
if(f[i]=='a') 
n1<- n1 + 1 else n2<- n2 + 1 # n1 counts a's n2 counts b's 
} 

[1] "b" "a" "a" "b" "a" "a" "a" NA NA NA NA NA NA NA NA NA NA NA NA NA 

Как вы можете видеть из номера результата a = 5 и b's = 2. Но когда я называю n1 и n2, он возвращает 4 для n1, что неверно. Почему это? Как я ошибаюсь? Заранее спасибо .

Я начинаю выборку a и b с вероятностью 0,5 .20 всего. В какой-то момент, о котором я не знаю заранее, количество a - число b превышает 2. Когда это произойдет, я хочу, чтобы вероятность выборки изменилась до 0,2. То, чего я пытаюсь достичь.

Answer 1

ifelse обозначен так, что не нужно использовать for-loop здесь. Это, как вы должны использовать:

set.seed(10) 
x <- runif(20) 
f <- ifelse(x<0.5, 'a', 'b') 

Затем используйте table сосчитать возникновение:

table(f) 
f 
a b 
13 7 

Чтобы получить n1:

n1 <- table(f)[['a']] 

EDIT OP хочет изменить образец согласно появление a и b. Вы должны использовать if и else не ifelse. Вы также должны предварительно выделить память для вектора f.

n1 <- 0 ; n2 <- 0 
N <- 20 
f <- vector(mode='character',N) 
set.seed(10) 
x<- runif(N) 
for(i in seq(N)) { 
    if (x[i] <0.5){ 
     f[i] <- 'a' 
     n1 <- n1 +1 
    } else { 
     f[i] <- 'b' 
     n2 <- n2 +1 
    } 
    if((n1-n2)==2) x <- runif(N) ## condition to regenerate sample 
}