r - 如何创建一个具有 3 个参数的函数，其中 1 个参数指定两列（在 r 中）？

Question

创建一个函数，该函数将数据框作为参数，一个参数指定要相乘的两列作为长度为 2 的数字向量，以及一个设置要创建的列的名称的参数。该函数应返回带有一列额外列的原始数据框。新列应包含两个指定列的乘积，并具有指定的名称。

之前执行的步骤：

set.seed(666)

produce <- data.frame(apples = rnorm(100), bananas = rnorm(100), carrots = rnorm(100))

标签： rfunction

Answer 1

dt_prod <- function(dt, cols, name) {
     dt[[name]] <- dt[[cols[1]]] * dt[[cols[2]]]
}