使用两个数据框的条件语句应用函数

Question

我有两个数据框 x 和 y。

x<-data.frame("j"=c("A","B","C"),"k"=c(1,90,14))
  x j  k
  1 A  1
  2 B 90
  3 C 14

y<-data.frame("A"=c(1,0,0,1,1),"B"=c(0,1,0,0,1),"C"=c(1,1,1,0,0))
  A B C
1 1 0 1
2 0 1 1
3 0 0 1
4 1 0 0
5 1 1 0

我需要一个带有条件语句的函数，如果在数据集 y 中有一个 0 a 替换为 -1 或 1 替换为 A 列中的 1 等等以获得这个结果。

z<-data.frame("A"=c(1,-1,-1,1,1),"B"=c(-90,90,-90,-90,90),"C"=c(14,14,14,-14,-14))
   A   B   C
1  1 -90  14
2 -1  90  14
3 -1 -90  14
4  1 -90 -14
5  1  90 -14

Answer 1

这是一个有趣的基础 R 中的单行代码，它使用了许多众所周知的函数。

setNames(do.call(data.frame, Map(function(x, y) x * c(-1, 1)[y+1], x$k, y)), LETTERS[1:3])
   A   B   C
1  1 -90  14
2 -1  90  14
3 -1 -90  14
4  1 -90 -14
5  1  90 -14

最里面的函数x * c(-1, 1)[y+1] 获取 x 的值（它是一个标量——1、90 或 14）并根据 y 的值将其乘以负数 -1 或 1。

Map 将 x$k 的值作为 x 并将 data.frame y 的变量作为 y 参数，对每一对执行函数并返回一个列表。三个向量。

此列表被转换为带有do.call 的data.frame，变量的名称由setNames 提供。