5

我有一个数据框z，我想根据两个旧列的值创建新列z。以下是过程：

>z<-cbind(x=1:10,y=11:20,t=21:30)
> z<-as.data.frame(z)
>z
    x  y  t
1   1 11 21
2   2 12 22
3   3 13 23
4   4 14 24
5   5 15 25
6   6 16 26
7   7 17 27
8   8 18 28
9   9 19 29
10 10 20 30

# 生成q等于列t乘以 4 的值的列，如果x=3和对于其他值x，它等于列的值t。

for (i in 1:nrow(z)){
  z$q[i]=if (z$x[i]==4) 4*z$t[i] else z$t[i]}

但是，我的问题是我想应用多个条件：

例如，我想得到这样的东西：

(If x=2, q=t*2; x=4, q=t*4; x=7, q=t*3; for other it is equal to t) 

> z
   x  y  t  q
1   1 11 21 21
2   2 12 22 44
3   3 13 23 23
4   4 14 24 96
5   5 15 25 25
6   6 16 26 26
7   7 17 27 81
8   8 18 28 28
9   9 19 29 29
10 10 20 30 30

如何使用循环或任何其他方法获得第二个输出？

4

8 回答 8

10

通过递归构建嵌套ifelse函数，您可以获得迄今为止提供的两种解决方案的好处：ifelse速度快并且可以处理任何类型的数据，而@Matthew 的解决方案功能更强大，但仅限于整数并且可能很慢。

decode <- function(x, search, replace, default = NULL) {

   # build a nested ifelse function by recursion
   decode.fun <- function(search, replace, default = NULL)
      if (length(search) == 0) {
         function(x) if (is.null(default)) x else rep(default, length(x))
      } else {
         function(x) ifelse(x == search[1], replace[1],
                                            decode.fun(tail(search, -1),
                                                       tail(replace, -1),
                                                       default)(x))
      }

   return(decode.fun(search, replace, default)(x))
}

请注意该decode函数是如何以 SQL 函数命名的。我希望像这样的函数可以用于基础 R 包......这里有几个例子说明了它的用法：

decode(x = 1:5, search = 3, replace = -1)
# [1]  1  2 -1  4  5
decode(x = 1:5, search = c(2, 4), replace = c(20, 40), default = 3)
# [1] 3 20  3  40  3

对于您的特定问题：

transform(z, q = decode(x, search = c(2,4,7), replace = c(2,4,3), default = 1) * t)

#    x  y  t  q
# 1   1 11 21 21
# 2   2 12 22 44
# 3   3 13 23 23
# 4   4 14 24 96
# 5   5 15 25 25
# 6   6 16 26 26
# 7   7 17 27 81
# 8   8 18 28 28
# 9   9 19 29 29
# 10 10 20 30 30
于 2013-01-05T03:08:50.193 回答
3

这是一个简单的解决方案，只需一个ifelse命令：

计算 的乘数t：

ifelse(z$x == 7, 3, z$x ^ (z$x %in% c(2, 4)))

完整的命令：

transform(z, q = t * ifelse(x == 7, 3, x ^ (x %in% c(2, 4))))

    x  y  t  q
1   1 11 21 21
2   2 12 22 44
3   3 13 23 23
4   4 14 24 96
5   5 15 25 25
6   6 16 26 26
7   7 17 27 81
8   8 18 28 28
9   9 19 29 29
10 10 20 30 30
于 2013-01-07T10:13:05.613 回答
3

生成一个多重向量：

tt <- rep(1, max(z$x))
tt[2] <- 2
tt[4] <- 4
tt[7] <- 3

这是您的新专栏：

> z$t * tt[z$x]
 [1] 21 44 23 96 25 26 81 28 29 30

> z$q <- z$t * tt[z$x]
> z
    x  y  t  q
1   1 11 21 21
2   2 12 22 44
3   3 13 23 23
4   4 14 24 96
5   5 15 25 25
6   6 16 26 26
7   7 17 27 81
8   8 18 28 28
9   9 19 29 29
10 10 20 30 30

如果 中有负值，这将不起作用z$x。

已编辑

这是上面的概括，其中一个函数用于生成乘数向量。实际上，我们根据参数创建了一个函数。

我们要转换以下值：

2 -> 2
4 -> 4
7 -> 3

否则，将采用默认值 1。

这是一个生成所需函数的函数：

f <- function(default, x, y) {
  x.min <- min(x)
  x.max <- max(x)
  y.vals <- rep(default, x.max-x.min+1)
  y.vals[x-x.min+1] <- y

  function(z) {
    result <- rep(default, length(z))
    tmp <- z>=x.min & z<=x.max
    result[tmp] <- y.vals[z[tmp]-x.min+1]
    result
  }
}

以下是我们如何使用它：

x <- c(2,4,7)
y <- c(2,4,3)

g <- f(1, x, y)

g是我们想要的功能。应该清楚的是，任何映射都可以通过x和y参数提供给f。

g(z$x)
## [1] 1 2 1 4 1 1 3 1 1 1

g(z$x)*z$t
## [1] 21 44 23 96 25 26 81 28 29 30

应该清楚这仅适用于整数值。

于 2012-12-31T20:15:56.850 回答
3

根据 Señor 的建议：

> z$q <- ifelse(z$x == 2, z$t * 2,
         ifelse(z$x == 4, z$t * 4,
         ifelse(z$x == 7, z$t * 3,
                          z$t * 1)))
> z
    x  y  t  q
1   1 11 21 21
2   2 12 22 44
3   3 13 23 23
4   4 14 24 96
5   5 15 25 25
6   6 16 26 26
7   7 17 27 81
8   8 18 28 28
9   9 19 29 29
10 10 20 30 30
于 2013-01-01T15:38:18.917 回答
2

你可以在

  • 碱基R
  • 用一根线
  • 其中映射非常清晰，可以在代码中阅读
  • 没有辅助函数（好的，一个匿名函数）
  • 方法适用于否定
  • 方法适用于任何原子向量（实数、字符）

像这样： 

> transform(z,q=t*sapply(as.character(x),function(x) switch(x,"2"=2,"4"=4,"7"=3,1)))
    x  y  t  q
1   1 11 21 21
2   2 12 22 44
3   3 13 23 23
4   4 14 24 96
5   5 15 25 25
6   6 16 26 26
7   7 17 27 81
8   8 18 28 28
9   9 19 29 29
10 10 20 30 30
于 2013-01-12T05:35:53.090 回答
2

我真的很喜欢在 florel 的博客上发布的“dinre”答案：

for (i in 1:length(data_Array)){
data_Array[i] <- switch(data_Array[i], banana="apple", orange="pineapple", "fig")
}

带有关于switch仔细阅读帮助页面以了解整数参数的警告。

于 2013-01-07T16:45:40.810 回答
1

这是decodeR 中用于字符向量（未经因子测试）的 SQL 版本，其操作与 SQL 版本一样。即它需要任意数量的目标/替换对，以及作为默认值的可选最后一个参数（请注意，默认值不会覆盖 NA）。

我可以看到它与dplyr'mutate操作一起非常有用。

> x <- c("apple","apple","orange","pear","pear",NA)

> decode(x, apple, banana)
[1] "banana" "banana" "orange" "pear"   "pear"   NA      

> decode(x, apple, banana, fruit)
[1] "banana" "banana" "fruit"  "fruit"  "fruit"  NA      

> decode(x, apple, banana, pear, passionfruit)
[1] "banana"       "banana"       "orange"       "passionfruit" "passionfruit" NA            

> decode(x, apple, banana, pear, passionfruit, fruit)
[1] "banana"       "banana"       "fruit"        "passionfruit" "passionfruit" NA

这是我正在使用的代码，我将在这里更新一个要点（链接）。

decode <- function(x, ...) {

  args <- as.character((eval(substitute(alist(...))))

  replacements <- args[1:length(args) %% 2 == 0]
  targets      <- args[1:length(args) %% 2 == 1][1:length(replacements)]

  if(length(args) %% 2 == 1)
    x[! x %in% targets & ! is.na(x)] <- tail(args,1)

  for(i in 1:length(targets))
    x <- ifelse(x == targets[i], replacements[i], x)

  return(x)

}
于 2014-11-19T06:20:58.440 回答
1

您也可以使用 match 来执行此操作。在将 col、pch 和 cex 等参数分配给散点图中的点时，我倾向于经常使用它

searchfor<-c(2,4,7)
replacewith<-c(2,4,3)

# generate multiplier column
# q could also be an existing vector where you want to replace certain entries
q<-rep(1,nrow(z))
#
id<-match(z$x,searchfor)
id<-replacewith[id]
# Apply the matches to q
q[!is.na(id)]<-id[!is.na(id)]
# apply to t
z$q<-q*z$t
于 2013-01-07T08:45:52.957 回答