在两个不同的部分计算功能

Question

我有一个向量y，它是一系列正整数。对于y中的每个值，我想计算：

• z <- 1/y，
• 但是对于y为1的元素，我需要设置z为零。
• 最后，我需要一个z的向量，y为0，1/y的{​​{1}}不为1.

我该怎么做？

Answer 1

使用ifelse并一步完成：

y <- 1:5
z <- ifelse(y==1, 0, 1/y)

cbind(y, z)
     y         z
[1,] 1 0.0000000
[2,] 2 0.5000000
[3,] 3 0.3333333
[4,] 4 0.2500000
[5,] 5 0.2000000

Answer 2

使用逻辑索引很容易做到这一点：

z <- 1/y  # will have some Inf values but not throw an error
z[y==0] <- 0   # changes the Inf's to 0

Answer 3

将Andrie所说的内容扩展到矩阵......

y <- matrix(1:25, ncol=5) # define the matrix y
z <- matrix(ifelse(y==1, 0, 1/y), ncol=ncol(y))

或者我可以说

z <- matrix(ifelse(y==1, 0, 1/y), ncol=5)