数据帧通过匹配行和列设置单元格值

Question

我在R中有一个数据帧，我想根据两个列表中的行和列的特定匹配来设置值。

以下是我必须遵循的流程。我有一个数据框，如下所示

   test.var1.t1  test.var2.t1   test.var1.t4   test.var3.t1   test.var3.t7
var1          0             0             0               0              0
var2          0             0             0               0              0
var3          0             0             0               0              0
var4          0             0             0               0              0

我有一个函数，它将列名作为输入，并返回一个与该列名关联的值。例如：一些_函数（“test.var1.t1”）可能返回10的值。

我有一些行名和列名的列表。我需要检查列表中的每个行名称和列名，将它们匹配在一起，计算匹配列的函数，并将值发布到相应的单元格中的数据框中。例如

行名称列表="var1""var2""var4"

列出col names=“test.var1.t1”“test.var2.t1”“test.var1.t4”“test.var3.t1”

var1应该与“test.var1.t1”和“test.var1.t4”匹配，所以我们从一些函数（）中获得“test.var1.t1”和“test.var1.t4”的函数值。假设这些值分别为10和20。

var2应该与“test.var2.t1”匹配，所以获取“test.var2.t1”的函数值。假设这个值是15。

var4没有匹配的列，var3也不在行列表中，所以这应该单独留在数据帧中。

完成后，数据帧应更改为：

      x.var1.t1   x.var2.t1   x.var1.t4   x.var3.t1   x.var3.t7
var1         10           0          20           0           0
var2          0          15           0           0           0
var3          0           0           0           0           0
var4          0           0           0           0           0

最好的方法是什么？

Answer 1

你可以这样做：

nn <- c("var1" ,"var2", "var4")
## I am using a for loop because the side effect is desired here
for(x in nn) {
  ## find the corresponding column names
  col <- grep(x,names(dx),value=TRUE)
  ## if any apply the function 
  if(length(col)>0) dx[x,col] <- get_val(x)
}

使用虚拟函数测试它get_val：

get_val <- function(x){
  switch(x,
         var1=c(15,20),
         var2=c(10))
} 

和dx：

dx <- read.table(text=" test.var1.t1  test.var2.t1   test.var1.t4   test.var3.t1   test.var3.t7
var1          0             0             0               0              0
var2          0             0             0               0              0
var3          0             0             0               0              0
var4          0             0             0               0              0",
           stringsAsFactors = FALSE)

你会得到预期的结果：

#      test.var1.t1 test.var2.t1 test.var1.t4 test.var3.t1 test.var3.t7
# var1           15            0           20            0            0
# var2            0           10            0            0            0
# var3            0            0            0            0            0
# var4            0            0            0            0            0