Fix daply bug

hadley · Apr 21, 2009 · 25902b2 · 25902b2
1 parent 5ba6a1e
commit 25902b2
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/NEWS b/NEWS
@@ -1,6 +1,7 @@
 plyr 0.1.8 (2008-XX-XX) ---------------------------------------------------
 
 * made rbind a little faster (~20%) using an idea from Richard Raubertas
+* daply now works correctly when splitting variables that contain empty factor levels
 
 plyr 0.1.7 (2008-04-15) ---------------------------------------------------
 

diff --git a/R/simplify-array.r b/R/simplify-array.r
@@ -42,7 +42,8 @@ list_to_array <- function(res, labels = NULL, .drop = FALSE) {
     in_labels <- list(NULL)
     in_dim <- n
   } else {
-    in_labels <- lapply(labels, unique)
+    in_labels <- lapply(labels, 
+      function(x) if(is.factor(x)) levels(x) else unique(x))
     in_dim <- sapply(in_labels, length)        
   }