Bug fixes for arrange

hadley · Apr 19, 2011 · d6a0ebe · d6a0ebe
1 parent 21fc841
commit d6a0ebe
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/NEWS b/NEWS
@@ -1,6 +1,8 @@
 Version 1.6
 ------------------------------------------------------------------------------
 
+* allow arrange to work with one column data frames
+
 Version 1.5.2
 ------------------------------------------------------------------------------
 

diff --git a/R/helper-arrange.r b/R/helper-arrange.r
@@ -16,7 +16,11 @@
 #' arrange(mtcars, cyl, desc(disp))
 arrange <- function(df, ...) {
   ord <- eval(substitute(order(...)), df, parent.frame())
-  unrowname(df[ord, ])
+  if(length(ord) != nrow(df)) {
+    stop("Length of ordering vectors don't match data frame size", 
+      call. = FALSE)
+  }
+  unrowname(df[ord, , drop = FALSE])
 }
 
 #' Descending order.