Зацикливание строк с использованием функции if-statement
Я новичок в R, и я пытаюсь суммировать 2 столбца заданного кадра данных, если оба элемента для суммирования удовлетворяют заданному условию. Чтобы все было ясно, я хочу сделать следующее:
> t.d<-as.data.frame(matrix(1:9,ncol=3))
> t.d
V1 V2 V3
1 4 7
2 5 8
3 6 9
> t.d$V4<-rep(0,nrow(t.d))
> for (i in 1:nrow(t.d)){
+ if (t.d$V1[i]>1 && t.d$V3[i]<9){
+ t.d$V4[i]<-t.d$V1[i]+t.d$V3[i]}
+ }
> t.d
V1 V2 V3 V4
1 4 7 0
2 5 8 10
3 6 9 0
Мне нужен эффективный код, так как мой реальный фреймворк имеет около 150000 строк и 200 столбцов. Это дает ошибку:
t.d$V4<-t.d$V1[t.d$V1>1]+ t.d$V3[t.d$V3>9]
Используется ли опция? Я пробовал это:
t.d<-as.data.frame(matrix(1:9,ncol=3))
t.d$V4<-rep(0,nrow(t.d))
my.fun<-function(x,y){
if(x>1 && y<9){
x+y}
}
t.d$V4<-apply(X=t.d,MAR=1,FUN=my.fun,x=t.d$V1,y=t.d$V3)
но он также дает ошибку.
Большое спасибо за вашу помощь.
Ответы
Ответ 1
Эта операция не требует циклов, приложений или операторов. Векторизованные операции и подмножество - это все, что вам нужно:
t.d <- within(t.d, V4 <- V1 + V3)
t.d[!(t.d$V1>1 & t.d$V3<9), "V4"] <- 0
t.d
V1 V2 V3 V4
1 1 4 7 0
2 2 5 8 10
3 3 6 9 0
Почему это работает?
На первом шаге я создаю новый столбец, который является прямой суммой столбцов V1 и V4. Я использую within как удобный способ обращения к столбцам d.f без необходимости писать d.f$V все время.
На втором шаге я подмножаю все строки, которые не соответствуют вашим условиям, и установите для них V4 значение 0.
Ответ 2
ifelse - ваш друг здесь:
t.d$V4<-ifelse((t.d$V1>1)&(t.d$V3<9), t.d$V1+ t.d$V3, 0)
Ответ 3
Я включу и поставлю еще одну версию. Поскольку вам нужен ноль, если условие не работает, а TRUE/FALSE - прославленные версии 1/0, просто выполняется умножение на условие:
t.d<-as.data.frame(matrix(1:9,ncol=3))
t.d <- within(t.d, V4 <- (V1+V3)*(V1>1 & V3<9))
... и это происходит быстрее, чем другие решения; -)
t.d <- data.frame(V1=runif(2e7, 1, 2), V2=1:2e7, V3=runif(2e7, 5, 10))
system.time( within(t.d, V4 <- (V1+V3)*(V1>1 & V3<9)) ) # 3.06 seconds
system.time( ifelse((t.d$V1>1)&(t.d$V3<9), t.d$V1+ t.d$V3, 0) ) # 5.08 seconds
system.time( { t.d <- within(t.d, V4 <- V1 + V3);
t.d[!(t.d$V1>1 & t.d$V3<9), "V4"] <- 0 } ) # 4.50 seconds
