替换R中向量中的值的自定义函数答案

【问题标题】：Custom function that replaces values in a vector in R替换R中向量中的值的自定义函数
【发布时间】：2018-08-17 03:12:39
【问题描述】：

我有这个代码：

val_hps_mix <- 0
co_neg_HP <- 1
co_pos_HP <- 2
e <- c(.1, -.2, 0, -.2)
if(val_hps_mix == 0 & co_neg_HP > 0 & co_pos_HP > 0) e[e < max(e[e < 0])] <- 0
if(val_hps_mix == 0 & co_neg_HP > 0 & co_pos_HP > 0) e[e < 0] <- e[e < 0]/length(e[e < 0])
e

正如预期的那样，这会返回：

[1]  0.1 -0.1  0.0 -0.1

如何将 IF 放入函数中？下面这个不起作用，我没有得到预期的结果..

    C.NAcc_hetero <- function(){
    if(val_hps_mix == 0 & co_neg_HP > 0 & co_pos_HP > 0) e[e < max(e[e < 0])] <- 0
    if(val_hps_mix == 0 & co_neg_HP > 0 & co_pos_HP > 0) e[e < 0] <- e[e < 0]/length(e[e < 0])
}

谢谢！

【问题讨论】：

标签： r function replace

【解决方案1】：

试试这个。如您所见，没有默认参数，因此您可以根据需要/想要更改和设置所有内容。

C.NAcc_hetero <- function(e, val_hps_mix, co_neg_HP, co_pos_HP){
  if(val_hps_mix == 0 & co_neg_HP > 0 & co_pos_HP > 0) e[e < max(e[e < 0])] <- 0
  if(val_hps_mix == 0 & co_neg_HP > 0 & co_pos_HP > 0) e[e < 0] <- e[e < 0]/length(e[e < 0])
  e
}
C.NAcc_hetero(e=c(.1, -.2, 0, -.2), val_hps_mix=0, co_neg_HP=1, co_pos_HP=2)
[1]  0.1 -0.1  0.0 -0.1

【讨论】：

@Erik Kornet 和@Jilber Urbina 谢谢！存储 e 的新值：e

【解决方案2】：

您可以通过在函数中输入val_hps_mix、co_neg_HP 和co_pos_HP 来做到这一点，如下所示：

e <- c(.1, -.2, 0, -.2)
C.NAcc_hetero <- function(val_hps_mix, co_neg_HP, co_pos_HP){
  if(val_hps_mix == 0 & co_neg_HP > 0 & co_pos_HP > 0) e[e < max(e[e < 0])] <- 0
  if(val_hps_mix == 0 & co_neg_HP > 0 & co_pos_HP > 0) e[e < 0] <- e[e < 0]/length(e[e < 0])
  return(e)
}

然后使用这些输入运行函数：

> C.NAcc_hetero(0, 1, 2)
[1]  0.1 -0.1  0.0 -0.1

使用这种结构，C.NAcc_hetero 的输出将根据val_hps_mix、co_neg_HP 和co_pos_HP 的值而变化

【讨论】：