如果满足条件,则用NA替换因子值

时间:2019-05-08 13:04:20

标签: r tidyverse forcats

我想根据另一列替换因子变量中的值,而又不更改初始因子水平。

示例:

x <- structure(list(Payee = structure(c(NA, 1L, 2L),
 .Label = c("0", "x"), class = "factor"), PayeeID_Hash = structure(c(NA, 1L,2L), 
.Label = c("0x31BCA02","0xB672841"), class = "factor")),
 row.names = c(NA,"tbl", "data.frame"))
> x
# A tibble: 3 x 2
  Payee PayeeID_Hash
  <fct> <fct>       
1 NA    NA          
2 0     0x31BCA02   
3 x     0xB672841  

Payee为'0'时,相应的PayeeID_Hash值应不存在(即应为NA)。请注意,我不想删除因子级别0x31BCA02(它将在Payee具有级别x的其他行中显示)。另外,我想保持PayeeID_Hash级别不变(我不想用其他值替换它们)。

预期输出:

> x
# A tibble: 3 x 2
  Payee PayeeID_Hash
  <fct> <fct>       
1 NA    NA          
2 0     NA          
3 x     0xB672841  

我可以通过将factor转换为character然后再转换为factor来做到这一点:

x %>%
  mutate(PayeeID_Hash = as.character(PayeeID_Hash),
         PayeeID_Hash = ifelse(Payee == "0", NA_character_, PayeeID_Hash),
         PayeeID_Hash = as.factor(PayeeID_Hash))

是否有另一种更清洁(即更直接)的方法?

1 个答案:

答案 0 :(得分:1)

我们可以使用replace并避免执行第2步和第4步。它将保持factor列不变,并且不会将factor强制转换为integer(除非已转换)转到character类),如ifelse

library(dplyr)
x %>%
   mutate(PayeeID_Hash = droplevels(replace(PayeeID_Hash, Payee == "0", NA)))
# A tibble: 3 x 2
#  Payee PayeeID_Hash
#  <fct> <fct>       
#1 <NA>  <NA>        
#2 0     <NA>        
#3 x     0xB672841