Question

我正在编写框架并尝试实现通用的init函数。

给定特定数据类型的向量v1：

如何使用与第一个向量v1中相同数据类型的NA值显式初始化另一个向量v2？

示例：

v1 <- c("a", "b")
v2 <- rep(NA, length(v1))  # would use the wrong type since NA is NA_logical_ by default
typeof(v2) # "logcial"

v2 <- rep(NA_character_)   # what I really want
typeof(v2) # "character"

v1 <- 1:2
v2 <- rep(NA_real_, length(v1))
typeof(v2) "double"

# ... same for NA_complex_, NA_integer_ ...

我是否真的必须编写一个函数并执行＆＃34;如果typeof（v1）则NA ...＆＃34; 或者是否有现有的功能或更聪明的方式？

Answer 1

The answers to this question might help you

@Josh O'Brien建议的可能解决方案：

rep(c(v1[0], NA), length(v1))

Answer 2

我想你可以自己动手。你必须输出所有内容，但只需要输入一次。

na_maker <- function(class){
  switch(class,
         double = NA_real_,
         integer = NA_integer_,
         character = NA_character_,
         logical = NA,
         complex = NA_complex_)
}


v1 <- c("a", "b")
v2 <- rep(na_maker(class(v1)), length(v1))

class(v2)
[1] "character"

您还可以创建另一个包装函数来简化操作：

rep_na <- function(vec){
  rep(na_maker(class(vec)), length(vec))
}

v2 <- rep_na(v1)

Answer 3

我的第一个想法（由你的评论加上）很简单：

v2 <- rep(NA, length(v1))
class(v2) <- class(v1)

单行将是：

v2 <- v1[v1==NA]

但这似乎比较慢，例如：

library(microbenchmark)                                                 
v1 <- sample(1:10000)                                                   

microbenchmark(                                                         
one = { v2 <- v1[v1==NA] },                                             
two = { v2 <- rep(NA, length(v1)); class(v1) <- class(v2)}, times = 1000
)                                                                       
#> Unit: microseconds
#>  expr    min     lq     mean  median      uq     max neval
#>   one 69.804 71.078 92.79524 71.6385 72.4335 650.255  1000
#>   two 18.983 19.424 22.96938 20.9525 21.5040 953.085  1000

具有与另一个向量

3 个答案: