R语言中基于函数分配的条件判断报错问题求助

错误原因 #

在R中，==运算符仅支持原子类型（数值、字符、逻辑值等）和列表的比较，而函数属于函数类型的对象，无法直接用==进行比较，这就是你遇到报错的原因。直接用字符串和函数对象比较也会因为类型不匹配触发同样错误。

可行的解决方法 #

方法1：用identical()直接比较函数对象 #

identical()可以准确判断两个对象是否完全一致，包括函数对象：

if (identical(gradient_fcn, binomial_grad)) {
  a = 1
} else {
  a = 2
}

注意：需要确保binomial_grad在当前嵌套函数的环境中是可访问的。

方法2：通过函数名称判断 #

获取gradient_fcn的名称，再和目标函数名字符串比较：

# 获取当前gradient_fcn的函数名
fcn_name = deparse(substitute(gradient_fcn))
if (fcn_name == "binomial_grad") {
  a = 1
} else {
  a = 2
}

或者直接读取函数对象的name属性：

fcn_name = attr(gradient_fcn, "name")
if (fcn_name == "binomial_grad") {
  a = 1
} else {
  a = 2
}

方法3：给函数添加自定义属性（推荐，扩展性更强） #

给每个梯度函数添加专属的类型属性，后续通过属性值判断：

# 定义梯度函数时添加属性
binomial_grad = function(x) {
  # 你的函数逻辑
}
attr(binomial_grad, "grad_type") = "binomial"

# 其他梯度函数同理
gaussian_grad = function(x) {
  # 你的函数逻辑
}
attr(gaussian_grad, "grad_type") = "gaussian"

# 判断时读取属性
if (attr(gradient_fcn, "grad_type") == "binomial") {
  a = 1
} else {
  a = 2
}

这种方式不受函数名称变化影响，后续新增梯度函数时只需添加对应属性即可，扩展性更好。

内容的提问来源于stack exchange，提问作者user19904