变量类是 data.table 中的列表

Question

假设我有以下数据：

library(data.table)
library(tidyverse)
Data <- 
  data.table(A = c("foo 32.3 56.2421 4.5 3.66", 
                 "365.11 9812.3 foooo 5.66 8.31",
                 ""))

有些行有四个数字，有些则没有。 我想要两个带有这些数字的新列。这是我的代码：

Data[, str_c("Number", 1:4) := str_extract_all(A, "\\d+\\.\\d") %>% transpose]

它工作得很好，但在第三行中，我在变量 Number1 到 Number2 中得到了 NULL（我想要 NA）。

真正的问题是Number1 到Number4 的类是list，但我想要character（在下一步中甚至更好numeric）。

我怎样才能做到这一点？我想使用带有提取和data.table 的代码。

Answer 1

您可以使用lapply 将值转换为数字，如果值不存在则返回NA，然后转置结果以创建两个新列。

library(data.table)

Data[, c("Number1", "Number2") := lapply(stringr::str_extract_all(A, "\\d+"), 
       function(x) if(length(x)) as.numeric(x) else NA) %>% transpose()]

Data

#                A Number1 Number2
#1:      foo 32 56      32      56
#2: 365 9812 foooo     365    9812
#3:                     NA      NA

---

使用tidyr 的extract 你可以这样做：

tidyr::extract(Data, A, c("Number1", "Number2"), 
              '(\\d+).*(\\d+)', remove = FALSE, convert = TRUE)

#                A Number1 Number2
#1:      foo 32 56      32       6
#2: 365 9812 foooo     365       2
#3:                     NA      NA