R代码中的多个条件

Question

假设我有一个数据集，

y <- c(0,0,1,2,2,1,0,1)
a <- c(0,2,1,1,0,2,0,3)
dat <- data.frame(y, a)

我想通过使用条件

来计算f <- digamma(a+y)的值

if(a>0 & y==0) then f
if(a==0 & y>0) then f
if (a==0 & y==0) then f = 1

如何使用R代码进行操作？

Answer 1

我认为这会给你事后的事情：

f <- digamma(a+y) 
f[a==0 & y==0] <- 1

或

f[is.nan(f)] <- 1

这是使用索引来查看向量中的哪些元素是nan，或者在第一种方法中a和y都是0.然后索引[]表示采用f的这些元素和赋值{{1}说这些元素是一个。