@ fabian-wernerが提供する答えは素晴らしいですが、オブジェクトは複数のクラスを持つことができ、「factor」は必ずしもによって返される最初のものではない可能性があるため、class(yes)
すべてのクラス属性をチェックするためにこの小さな変更を提案します。
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)
}
また、保持する属性のユーザー選択に基づいてbase :: ifelse()が属性を保持するように文書化されたオプションを追加するように、R開発チームにリクエストを送信しました。リクエストはこちら:https : //bugs.r-project.org/bugzilla/show_bug.cgi?id=16609 //bugs.r-project.org/bugzilla/show_bug.cgi?id=16609-それは常に現在の方法であったという理由ですでに「WONTFIX」としてフラグが付けられています、しかし、単純な追加で多くのRユーザーの頭痛を軽減できる理由について、フォローアップの引数を提供しました。おそらく、そのバグスレッドでの「+1」は、Rコアチームの再検討を促すでしょう。
編集:これは、ユーザーがどの属性を保持するかを指定できる、より優れたバージョンです。「cond」(デフォルトはifelse()の動作)、「yes」、上記のコードによる動作、または「no」の場合「いいえ」の値の属性の方が優れています。
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()
しifelse
ながら代替できるdplyrパッケージに関数が追加されました - 最近の回答として以下に掲載されています。(このコメントの時点で)他の多くの回答が上位にランク付けされた他の多くの回答とは異なり、ユニットテストおよびドキュメント化された機能をCRANパッケージに提供することでこの問題を解決するため、ここで注目します。