Передача меток в xlab и ylab в ggplot2 - PullRequest
0 голосов
/ 13 февраля 2019

Я создал функцию, цель которой - создать серию графиков векторизованным способом.Функции частично выполняют то, что я хочу, это обновление, обновление графика на основе выбранных переменных.Однако я не могу передать аргумент метки (т. Е. Label_x и label_y), чтобы xlab и ylab обновлялись последовательно.

library(tidyverse)
plot_scatter_with_label <- function(df,
                                    var_x,
                                    var_y,
                                    label_x,
                                    label_y,
                                    geom_smooth = FALSE,
                                    point_shape = 16,
                                    point_color = "#EB3300",
                                    point_size = 1,
                                    point_alpha = 1,
                                    smooth_method = "loess",
                                    smooth_se = FALSE,
                                    smooth_color = "navy") {
  df <- data.frame(lapply(df, function(x) as.numeric(as.character(x))))
  scatter_plot <- function(x, y) {
    p <- ggplot(df, aes_string(x = x, y = y)) + 
      geom_point(shape = point_shape, color = point_color, size = point_size, alpha = point_alpha) + 
      ylab(label_y) + xlab(label_x)
    p
  }
  map2(
    var_y, label_y,
    ~ map(var_x, scatter_plot, y = .x)
  )
}

Пример

plot_scatter_with_label(
  df = mtcars,
  var_y = c("mpg", "hp"),
  label_y = c("Miles per gallon [Mpg]", "Horse power [CV]"),
  var_x = c("cyl", "gear"),
  label_x = c("Cylinders [n]", "Gear [n]")
)

Я ожидалполучить следующие графики:

1) миль на галлон против цил

2) миль на галлон против редуктора

3) л.с. против цил

4) л.с. против редуктора

Похоже, что я получил эти 4 графика, но ярлыки не обновляются, как ожидалось.Он всегда возвращает первый аргумент, определенный в label_x и label_y.

Любая помощь будет принята с благодарностью.

С уважением,

1 Ответ

0 голосов
/ 15 февраля 2019

Мы можем использовать pmap или pwalk для передачи данных в plot_scatter_with_label функцию

library(tidyverse)

plot_scatter_with_label <- function(dat,
                                    var_x,
                                    var_y,
                                    label_x,
                                    label_y,
                                    geom_smooth = FALSE,
                                    point_shape = 16,
                                    point_color = "#EB3300",
                                    point_size = 1,
                                    point_alpha = 1,
                                    smooth_method = "loess",
                                    smooth_se = FALSE,
                                    smooth_color = "navy") {

  if (is.character(var_x)) {
    print('character column names supplied, use rlang::sym()')
    var_x <- rlang::sym(var_x)
  } else {
    print('bare column names supplied, use dplyr::enquo()')
    var_x <- enquo(var_x)
  }

  if (is.character(var_y)) {
    var_y <- rlang::sym(var_y)
  } else {
    var_y <- enquo(var_y)
  }

  p <- ggplot(dat, aes(x = !! var_x, y = !! var_y)) + 
    geom_point(shape = point_shape, color = point_color, 
               size = point_size, alpha = point_alpha) + 
    ylab(label_y) + 
    xlab(label_x) +
    ggtitle(paste0(label_x, " ~ ", label_y))
  print(p)

}

Создать фрейм данных, чтобы мы могли проходить по каждой строке и столбцу

var_y = c("mpg", "hp")
label_y = c("Miles per gallon [Mpg]", "Horse power [CV]")
var_x = c("cyl", "gear")
label_x = c("Cylinders [n]", "Gear [n]")

var_xy <- expand.grid(var_x, var_y, stringsAsFactors = FALSE)
label_xy <- expand.grid(label_x, label_y, stringsAsFactors = FALSE)
select_dat <- data.frame(var_xy, label_xy, stringsAsFactors = FALSE)
str(select_dat)

#> 'data.frame':    4 obs. of  4 variables:
#>  $ Var1  : chr  "cyl" "gear" "cyl" "gear"
#>  $ Var2  : chr  "mpg" "mpg" "hp" "hp"
#>  $ Var1.1: chr  "Cylinders [n]" "Gear [n]" "Cylinders [n]" "Gear [n]"
#>  $ Var2.1: chr  "Miles per gallon [Mpg]" "Miles per gallon [Mpg]" "Horse power [CV]" "Horse power [CV]"

Передать каждый ряд функции plot_scatter_with_label

pwalk(select_dat, ~ plot_scatter_with_label(mtcars, ..1, ..2, ..3, ..4))

#> [1] "character column names supplied, use rlang::sym()"

#> [1] "character column names supplied, use rlang::sym()"

#> [1] "character column names supplied, use rlang::sym()"

#> [1] "character column names supplied, use rlang::sym()"

Создано в 2019-02-14 с помощью представительного пакета (v0.2.1.9000)

...