Browse code

update facet_labeller

Guangchuang Yu authored on 11/09/2018 05:49:47
Showing 1 changed files

... ...
@@ -36,8 +36,13 @@ facet_plot <- function(p, panel, data, geom, mapping=NULL, ...) {
36 36
 ##' @export
37 37
 ##' @author Guangchuang Yu
38 38
 facet_labeller <- function(p, label) {
39
-    p+ facet_grid( . ~ .panel, scales="free_x",
40
-               labeller = labeller(.panel = label))
39
+    lbs  <-  levels(p$data$.panel)
40
+    names(lbs)  <-  lbs
41
+    label <- label[names(label) %in% lbs]
42
+    lbs[names(label)]  <-  label
43
+
44
+    p + facet_grid( . ~ .panel, scales="free_x",
45
+               labeller = labeller(.panel = lbs))
41 46
 }
42 47
 
43 48
 ##' @importFrom ggplot2 facet_grid