0

我目前有这个函数，它接受一个表和两个表达式列表，并对它们求值，将它们变成两个矩阵。我使用两个列表而不是...因为我需要能够确定表达式是落在 Y 还是 Z 中。

func = function(tb, Y, Z) {
  Y_matrix = enquo(Y) %>% 
    eval_tidy(tb) %>%
    as.data.frame %>%
    as.matrix

  Z_matrix = enquo(Z) %>%
    eval_tidy(tb) %>%
    as.data.frame %>%
    as.matrix

  list(Y_matrix, Z_matrix)
}

例如：

> tb = data.frame(a = 1:3, b = 3:1, c = 2*(1:3))
> func(tb, list(a + b, a - b), list(a*c, a + b + c))
[[1]]
     c.4L..4L..4L. c..2L..0L..2L.
[1,]             4             -2
[2,]             4              0
[3,]             4              2

[[2]]
     c.2..8..18. c.6..8..10.
[1,]           2           6
[2,]           8           8
[3,]          18          10

但是，我还想将列表中的表达式捕获为文本，因此我可以使用它们来命名列。例如，我想要的输出是这样的：

> func(tb, list(a + b, a - b), list(a*c, a + b + c))
[[1]]
             a+b            a-b
[1,]             4             -2
[2,]             4              0
[3,]             4              2

[[2]]
              a*c         a + b + c
[1,]           2           6
[2,]           8           8
[3,]          18          10

我怎样才能做到这一点？

4

2 回答 2

3

您可以使用类似deparse检查列表中的语言元素将它们转换为字符串的方法。例如

func = function(tb, Y, Z) {
  Y_names <- map_chr(as.list(enexpr(Y))[-1], deparse)
  Y_matrix = enquo(Y) %>% 
    eval_tidy(tb) %>%
    as.data.frame %>%
    set_names(Y_names) %>% 
    as.matrix

  Z_names <- map_chr(as.list(enexpr(Z))[-1], deparse)
  Z_matrix = enquo(Z) %>%
    eval_tidy(tb) %>%
    as.data.frame %>%
    set_names(Z_names) %>% 
    as.matrix

  list(Y_matrix, Z_matrix)
}

这适用于这个例子

func(tb, list(a + b, a - b), list(a*c, a + b + c))
# [[1]]
#      a + b a - b
# [1,]     4    -2
# [2,]     4     0
# [3,]     4     2

# [[2]]
#      a * c a + b + c
# [1,]     2         6
# [2,]     8         8
# [3,]    18        10
于 2018-05-10T19:12:15.460 回答
2

我们可以将参数作为 quosure 传递，然后进行评估 ( !!!)

library(tidyverse)
f1 <- function(tb, Y, Z) {
       Y_matrix <-  tb %>% 
             transmute(!!! Y)

       Z_matrix <- tb %>%
             transmute(!!! Z)

       list(Y_matrix, Z_matrix)


   }

f1(tb, quos(a + b, a - b), quos(a*c, a + b + c))
#[[1]]
#  a + b a - b
#1     4    -2
#2     4     0
#3     4     2

#[[2]]
#  a * c a + b + c
#1     2         6
#2     8         8
#3    18        10
于 2018-05-10T19:08:32.423 回答