Looking for a ifelse hiệu quả hơn()

Question

Trong khi R của ifelse là vô cùng tiện dụng, nó có một nhược điểm cụ thể: trong cuộc gọi ifelse(test, yes, no) tất cả các yếu tố của yes và no được đánh giá, ngay cả những người đó sẽ được bỏ đi. Điều này là khá lãng phí nếu bạn đang sử dụng nó ở giữa một bài tập số phức tạp, nói trong một chức năng sẽ được cấp cho integrate, uniroot, optim hoặc bất cứ điều gì. Ví dụ, người ta có thể có

ifelse(test, f(x, y, z), g(a, b, c)) 

nơi f và g là tùy tiện chức năng phức tạp hoặc chậm, có thể liên quan đến lồng nhau hơn ifelse 's.

Có ai đã viết thay thế cho ifelse chỉ đánh giá các thành phần của yes/no sẽ được giữ lại không? Về cơ bản, điều gì đó dọc theo các dòng của

out <- test 
for(i in seq_along(out)) 
{ 
    if(test[i]) out[i] <- f(x[i], y[i], z[i]) 
    else out[i] <- g(a[i], b[i], c[i]) 
} 

nhưng không có sự vụng về/không hiệu quả của vòng lặp rõ ràng. Điều này thậm chí có thể mà không đi vào bên trong của R?

Answer 1

Tôi không nghĩ rằng vấn đề là ifelse. f và g chỉ được đánh giá một lần trong biểu thức của bạn. Tôi nghĩ rằng vấn đề của bạn là f và g chậm với vectơ.

Bạn có thể thay đổi các cuộc gọi thành f và g để chúng chỉ được đánh giá trên một tập hợp con của vectơ.

out <- numeric(length(test)) #or whatever the output type is 
out[test] <- f(x[test], y[test], z[test]) 
out[!test] <- g(x[!test], y[!test], z[!test]) 

Bạn cần điều chỉnh điều này nếu có bất kỳ thành phần nào của test là NA.