R：从调用函数的函数计算新的数据框列

Question

我正在尝试使用多参数计算数据框的新列 功能。当该功能很简单时，一切都很好。当那个 函数调用另一个函数，事情出错了。

第 1 部分：感兴趣的数据集。

# The goal is to transform Foo into a new column.
df <- data.frame(
   Year =c(1901,1901,1900,1902),
   Month=c(   2,   1,   2,   1),
   Foo  =c(   1,   2,   3,   4)
)

# This simple multi-arg transform works fine.
Foo2Baz <- function(year, month, foo) {
  return(year + month + foo)
}
df$Baz <- Foo2Baz(df$Year, df$Month, df$Foo)
# Expected Baz column: [1904, 1904, 1905, 1907] 
df$Baz
[1] 1904 1904 1905 1907

第 2 部分：查找表

失败的更有趣的转换就像上面的 Foo2Baz， 除了它会调用另一个函数来查找某个值 其他表。

# First we load a dataframe that backs the lookup operation. It's nice
# to keep as a dataframe, as opposed to matrix, for human inspection
# of cells.
lookup_table <- data.frame(
  Year=c(1900, 1901, 1902),
  Jan =c(  10,   20,   30),
  Feb =c( 100,  200,  200))

# Then we define a function to lookup a cell.
Lookup <- function(year, month) {
  return(lookup_table[
    year - 1899, # converts year to row index
    1 + month    # converts month to column, skipping 'Year'
  ])
}
# We expect a lookup of 1901-Feb to be 200
Lookup(1901, 2)
[1] 200

第 3 部分：调用函数的转换函数。

# The goal is to transform Foo into new column Bar
# by looking up that case's Year and Month.
Foo2Bar <- function(year, month, foo) {
  return(foo + Lookup(year, month))
}

# We expect case 1 (1901,Feb,Foo=1) to have Bar=201
Foo2Bar(1901,2,1)
[1] 201

# We expect case 4 (1902,Jan,Foo=4) to have Bar=34
Foo2Bar(1902,1,4)
[1] 34

第 4 部分：如何计算 Bar 的新列？

似乎我们现在可以使用 Foo2Bar 计算 Bar 列，就像 我们使用了更简单的 Foo2Baz：

df$Bar <- Foo2Bar(df$Year, df$Month, df$Foo)
df$Bar
    Feb Jan Feb.1 Jan.1
2   201  21   201    21
2.1 202  22   202    22
1   103  13   103    13
3   204  34   204    34

没有专栏？取而代之的是一个矩阵，其中行和列看起来像我们发送给 Lookup() 函数的各种输入？！

我尝试了 do.call、apply、lapply、sapply 和 dplyr mutate 的变体。我似乎只是在这里遗漏了一些基本的东西。

Answer 1

我们只需要cbind

Lookup <- function(year, month) {
   lookup_table[cbind(
           year - 1899, 
             1 + month)   
     ]
 }

Lookup(1901, 2)
#[1] 200
Foo2Bar(1901,2,1)
#[1] 201

df$Bar <- Foo2Bar(df$Year, df$Month, df$Foo)
df
#  Year Month Foo Bar
#1 1901     2   1 201
#2 1901     1   2  22
#3 1900     2   3 103
#4 1902     1   4  34