是否有像switch这样的函数在dplyr :: mutate中运行?

时间:2015-02-18 20:27:53

标签: r dplyr

我不能在mutate内部使用switch,因为它返回整个向量而不是行。作为一个黑客,我正在使用:

pick <- function(x, v1, v2, v3, v4) {
    ifelse(x == 1, v1,
           ifelse(x == 2, v2,
                  ifelse(x == 3, v3,
                         ifelse(x == 4, v4, NA))))
}

这在mutate内部起作用,现在很好,因为我通常选择4件事,但这可能会改变。你能推荐一个替代方案吗?

例如:

library(dplyr)
df.faithful <- tbl_df(faithful)
df.faithful$x  <- sample(1:4, 272, rep=TRUE)
df.faithful$y1 <- rnorm(n=272, mean=7, sd=2)
df.faithful$y2 <- rnorm(n=272, mean=5, sd=2)
df.faithful$y3 <- rnorm(n=272, mean=7, sd=1)
df.faithful$y4 <- rnorm(n=272, mean=5, sd=1)

使用pick

mutate(df.faithful, y = pick(x, y1, y2, y3, y4))
Source: local data frame [272 x 8]

   eruptions waiting x        y1        y2       y3       y4        y
1      3.600      79 1  8.439092 5.7753006 8.319372 5.078558 8.439092
2      1.800      54 2 13.515956 6.1971512 6.343157 4.962349 6.197151
3      3.333      74 4  7.693941 6.8973365 5.406684 5.425404 5.425404
4      2.283      62 4 12.595852 6.9953995 7.864423 3.730967 3.730967
5      4.533      85 3 11.952922 5.1512987 9.177687 5.511899 9.177687
6      2.883      55 3  7.881350 1.0289711 6.304004 3.554056 6.304004
7      4.700      88 4  8.636709 6.3046198 6.788619 5.748269 5.748269
8      3.600      85 1  8.027371 6.3535056 7.152698 7.034976 8.027371
9      1.950      51 1  5.863370 0.1707758 5.750440 5.058107 5.863370
10     4.350      85 1  7.761653 6.2176610 8.348378 1.861112 7.761653
..       ...     ... .       ...       ...      ...      ...      ...

我们看到如果x == 1,我将y1中的值复制到y中,依此类推。这就是我想要做的,但我希望能够做到这一点,无论我有4或400列的列表。

尝试使用switch

mutate(df.faithful, y = switch(x, y1, y2, y3, 4))

Error in switch(c(1L, 2L, 4L, 4L, 3L, 3L, 4L, 1L, 1L, 1L, 4L, 3L, 1L,  : 
EXPR must be a length 1 vector

尝试使用list

mutate(df.faithful, y = list(y1, y2, y3, y4)[[x]])
Error in list(c(8.43909205142925, 13.5159559591257, 7.69394050059568,  : 
recursive indexing failed at level 2

尝试使用c

mutate(df.faithful, y = c(y1, y2, y3, y4)[x])
Source: local data frame [272 x 8]

   eruptions waiting x        y1        y2       y3       y4         y
1      3.600      79 1  8.439092 5.7753006 8.319372 5.078558  8.439092
2      1.800      54 2 13.515956 6.1971512 6.343157 4.962349 13.515956
3      3.333      74 4  7.693941 6.8973365 5.406684 5.425404 12.595852
4      2.283      62 4 12.595852 6.9953995 7.864423 3.730967 12.595852
5      4.533      85 3 11.952922 5.1512987 9.177687 5.511899  7.693941
6      2.883      55 3  7.881350 1.0289711 6.304004 3.554056  7.693941
7      4.700      88 4  8.636709 6.3046198 6.788619 5.748269 12.595852
8      3.600      85 1  8.027371 6.3535056 7.152698 7.034976  8.439092
9      1.950      51 1  5.863370 0.1707758 5.750440 5.058107  8.439092
10     4.350      85 1  7.761653 6.2176610 8.348378 1.861112  8.439092
..       ...     ... .       ...       ...      ...      ...       ...

不会产生错误,但行为不符合预期。

9 个答案:

答案 0 :(得分:23)

对于OP来说太晚了,但是如果这出现在搜索中......

dplyr v0.5有recode()switch()的矢量化版本,所以

data_frame(
  x = sample(1:4, 10, replace=TRUE),
  y1 = rnorm(n=10, mean=7, sd=2),
  y2 = rnorm(n=10, mean=5, sd=2),
  y3 = rnorm(n=10, mean=7, sd=1),
  y4 = rnorm(n=10, mean=5, sd=1)
) %>%
mutate(y = recode(x,y1,y2,y3,y4))

按预期产生:

# A tibble: 10 x 6
       x        y1       y2       y3       y4        y
   <int>     <dbl>    <dbl>    <dbl>    <dbl>    <dbl>
1      2  6.950106 6.986780 7.826778 6.317968 6.986780
2      1  5.776381 7.706869 7.982543 5.048649 5.776381
3      2  7.315477 2.213855 6.079149 6.070598 2.213855
4      3  7.461220 5.100436 7.085912 4.440829 7.085912
5      3  5.780493 4.562824 8.311047 5.612913 8.311047
6      3  5.373197 7.657016 7.049352 4.470906 7.049352
7      2  6.604175 9.905151 8.359549 6.430572 9.905151
8      3 11.363914 4.721148 7.670825 5.317243 7.670825
9      3 10.123626 7.140874 6.718351 5.508875 6.718351
10     4  5.407502 4.650987 5.845482 4.797659 4.797659

(也适用于命名的args,包括字符和因子x&#39; s。

答案 1 :(得分:4)

您现在可以将dplyr的功能case_whenmutate()一起使用。

按照您的示例生成数据:

library(dplyr)

df.faithful <- tbl_df(faithful)
df.faithful$x  <- sample(1:4, 272, rep=TRUE)
df.faithful$y1 <- rnorm(n=272, mean=7, sd=2)
df.faithful$y2 <- rnorm(n=272, mean=5, sd=2)
df.faithful$y3 <- rnorm(n=272, mean=7, sd=1)
df.faithful$y4 <- rnorm(n=272, mean=5, sd=1)

现在,我们使用pick()定义一个新的case_when函数:

pick2 <- function(x, v1, v2, v3, v4) {
  out = case_when(
    x == 1 ~ v1,
    x == 2 ~ v2,
    x == 3 ~ v3,
    x == 4 ~ v4
  )
  return(out)
}

您会发现您可以在mutate()内完美使用它:

df.faithful %>% 
  mutate(y = pick2(x, y1, y2, y3, y4))

输出为:

# A tibble: 272 x 8
   eruptions waiting     x    y1    y2    y3    y4     y
       <dbl>   <dbl> <int> <dbl> <dbl> <dbl> <dbl> <dbl>
 1      3.6       79     3  8.73  7.23  8.89  4.04  8.89
 2      1.8       54     3  9.97  4.31  7.06  5.05  7.06
 3      3.33      74     1  6.65  7.23  4.46  6.49  6.65
 4      2.28      62     1  6.40  4.39  5.41  3.49  6.40
 5      4.53      85     4  3.96  8.85  7.43  6.51  6.51
 6      2.88      55     4  6.36  8.08  5.82  5.06  5.06
 7      4.7       88     1  5.91  6.47  6.43  5.88  5.91
 8      3.6       85     1  7.77  4.55  6.56  5.05  7.77
 9      1.95      51     4  5.74  6.46  6.95  4.26  4.26
10      4.35      85     1  7.04  1.73  5.71  2.53  7.04
# ... with 262 more rows

答案 2 :(得分:3)

x的每个值执行操作。这是data.table版本,我认为可以在dplyr中完成类似的事情:

library(data.table)

dt = data.table(x = c(1,1,2,2), a = 1:4, b = 4:7)

dt[, newcol := switch(as.character(x), '1' = a, '2' = b, NA), by = x]
dt
#   x a b newcol
#1: 1 1 4      1
#2: 1 2 5      2
#3: 2 3 6      6
#4: 2 4 7      7

答案 3 :(得分:2)

您可以通过以下方式修改您的功能:

map <- data.frame(i=1:2,v=10:11)
#   i  v
# 1 1 10
# 2 2 11

set.seed(1)
x <- sample(1:3,10,rep=T)
#  [1] 1 2 2 3 1 3 3 2 2 1

i <- match(x,map$i)
ifelse(is.na(i),x,map$v[i])
# [1] 10 11 11  3 10  3  3 11 11 10

我们的想法是将您要查找的值和替换值保存在单独的数据框map中,然后使用match来匹配x和{{ 1}}。

<强> [更新]

您可以将此解决方案汇总到可在map中使用的函数:

mutate

答案 4 :(得分:2)

这是使用data.table的另一种方式。我们的想法是基本上使用组合创建 data.table,然后执行 join ,如下所示:

我将使用@ eddi的答案中的data.table。

require(data.table)
key = data.table(x = 1:2, col = c("a", "b"))

setkey(dt, x)
dt[key, new_col := get(i.col), by=.EACHI]
#    x a b new_col
# 1: 1 1 4       1
# 2: 1 2 5       2
# 3: 2 3 6       6
# 4: 2 4 7       7

join 在列x上执行。对于的每一行,都会找到 dt 中相应的匹配行。例如: key 中的x = 1 dt 的行 1 2 匹配。在这些行上,我们会访问存储在 col中的列,即&#34; a&#34;。 get("a")会为匹配的行返回列a的值,即 1 2 。希望这会有所帮助。

by=.EACHI确保为new_col := get(i.col)中的每一行评估表达式key。您可以详细了解here

答案 5 :(得分:1)

备用(更多参与)路线涉及使用tidyr

df %>%
  mutate(row = row_number()) %>%
  gather(n, y, y1:y4) %>%
  mutate(n = as.integer(str_extract(n, "[0-9]+"))) %>%
  filter(x == n) %>%
  arrange(row) %>%
  select(-c(row, n))

答案 6 :(得分:1)

我有点晚了,但这是我使用mapply的解决方案。

vswitch <- function(x, ...) {
  mapply(FUN = function(x, ...) {
           switch(x, ...)
         }, x, ...)
}

mutate(df.faithful, y = vswitch(x, y1, y2, y3, y4))

答案 7 :(得分:1)

如果您想在class SampleRepository { fun getInstance(): SampleRepository { // return instance of SampleRepository } fun getDataFromService(): LiveData<Type> { // start some service and return LiveData } } class SampleViewModel { private val sampleRepository = SampleRepository.getInstance() private var sampleLiveData = MutableLiveData<Type>() // getter for sampleLiveData fun getSampleLiveData(): LiveData<Type> = sampleLiveData fun startService() { sampleLiveData.postValue(sampleRepository.getDataFromService()) } } class SampleView { private var sampleViewModel: SampleViewModel // for activities, this SampleMethod is often their onCreate() method fun SampleMethod() { // instantiate sampleViewModel sampleViewModel = ViewModelProviders.of(this).get(SampleViewModel::class.java) // observe LiveData of sampleViewModel sampleViewModel.getSampleLiveData().observe(viewLifecycleOwner, Observer<Type> { newData -> // update UI here using newData } } 中使用switch,则必须先执行mutate

rowwise

答案 8 :(得分:1)

比user6702291建议的解决方案更复杂的版本是使用map函数,例如map_dbl()。它更复杂,但我认为值得分享,因为它在其他情况下仍具有通用性,在这种情况下,您要使用的功能还没有矢量化版本。

在这种情况下,它将像这样工作。

 tibble.faithful %>% 
  mutate(y = map_dbl(seq_along(x), ~switch(x[.x], y1, y2, y3, y4)[1]))

我实际上不确定为什么需要“ [1]”,但我还是想作为建议分享。