R list comprehension if else

Question

我来自Python的世界。 我想只得到正数并将非正数设为零。 在Python中：

>> a = [1,2,3,-1,-2, 0,1,-9]
>> [elem if elem>0 else 0 for elem in a]
[1, 2, 3, 4, 0, 0, 0, 1, 0]

假设我在R中有一个向量，我怎样才能得到相同的结果。

a <- c(1,2,3,-1,-2, 0,1,-9)

Answer 1

使用ifelse

> ifelse(a>0, a, 0)
[1] 1 2 3 0 0 0 1 0

请参阅?ifelse了解详情。

您还可以使用[选择符合条件（a<=0）的值并将其替换为0

> a[a<=0] <- 0
> a
[1] 1 2 3 0 0 0 1 0

请参阅?"["。

a<=0将为您提供一个布尔向量，您可以将其用作索引来识别小于或等于0的值，然后执行所需的替换。

尽管使用[或ifelse是最常见的，但在此我还提供了其他一些选择：

a[which(a<=0)] <- 0       
#----------------------
a[a %in% 0:min(a)] <- 0    # equivalent to a[!a %in% 0:max(a)] <- 0 
#----------------------
match(a, 1:max(a), nomatch=0 )
#----------------------
pmax(a, 0)