@ Fabian-werner tarafından verilen cevap harika, ama nesneler birden fazla sınıfa sahip olabilir ve "faktör" illa döndürülmeyebilir class(yes)
, bu yüzden tüm sınıf özelliklerini kontrol etmek için bu küçük değişikliği öneriyorum:
safe.ifelse <- function(cond, yes, no) {
class.y <- class(yes)
if ("factor" %in% class.y) { # Note the small condition change here
levels.y = levels(yes)
}
X <- ifelse(cond,yes,no)
if ("factor" %in% class.y) { # Note the small condition change here
X = as.factor(X)
levels(X) = levels.y
} else {
class(X) <- class.y
}
return(X)
}
Ben de R :: Geliştirme ekibi ile temel :: ifelse () korumak için hangi özellikleri korumak için kullanıcı seçimi dayalı nitelikleri korumak için belgelenmiş bir seçenek eklemek için bir istek gönderdim. İstek burada: https://bugs.r-project.org/bugzilla/show_bug.cgi?id=16609 - Her zaman olduğu gibi olduğu için zaten "WONTFIX" olarak işaretlenmiş, ancak basit bir eklemenin neden birçok R kullanıcısının baş ağrısını kurtarabileceğine dair bir takip argümanı sundum. Belki de bu hata dizisindeki "+1", R Core ekibini ikinci bir bakış atmaya teşvik edecektir.
DÜZENLEME: Burada, kullanıcının "cond" (varsayılan ifelse () davranışı), "yes", yukarıdaki koda göre davranışı veya "no" ifadesini, "hayır" değerinin özellikleri daha iyidir:
safe_ifelse <- function(cond, yes, no, preserved_attributes = "yes") {
# Capture the user's choice for which attributes to preserve in return value
preserved <- switch(EXPR = preserved_attributes, "cond" = cond,
"yes" = yes,
"no" = no);
# Preserve the desired values and check if object is a factor
preserved_class <- class(preserved);
preserved_levels <- levels(preserved);
preserved_is_factor <- "factor" %in% preserved_class;
# We have to use base::ifelse() for its vectorized properties
# If we do our own if() {} else {}, then it will only work on first variable in a list
return_obj <- ifelse(cond, yes, no);
# If the object whose attributes we want to retain is a factor
# Typecast the return object as.factor()
# Set its levels()
# Then check to see if it's also one or more classes in addition to "factor"
# If so, set the classes, which will preserve "factor" too
if (preserved_is_factor) {
return_obj <- as.factor(return_obj);
levels(return_obj) <- preserved_levels;
if (length(preserved_class) > 1) {
class(return_obj) <- preserved_class;
}
}
# In all cases we want to preserve the class of the chosen object, so set it here
else {
class(return_obj) <- preserved_class;
}
return(return_obj);
} # End safe_ifelse function
if_else()
dplyr paketindeifelse
doğru Date nesnesi sınıflarını korurken yerine kullanılabilecek bir işlev var - aşağıda yeni bir yanıt olarak yayınlandı . Burada dikkat çekiyorum, çünkü bu sorunu çözdüğü gibi bir CRAN paketinde birim test edilmiş ve belgelenmiş bir fonksiyon sağlayarak (bu yorumdan itibaren) önde sıralanan diğer cevapların aksine bu sorunu çözüyor.