Question

我有一个包含字符观测值的数据表：

library(data.table)
library(stringr)

DT = data.table(strings = c('AAABD', 'BBDA', 'AACBDAA', 'ABACD'))

我想创建一个变量，该变量在每个观察结果中均包含一个计数值“ A”，“ AA”和“ AAA”作为列表。为此，我创建了一个函数foo：

foo <- function(str) {
    n <- str_count(str, 'A')
    n2 <- str_count(str, 'AA')
    n3 <- str_count(str, 'AAA')
    df <- list('n' = n, 'n2' = n2, 'n3' = n3)

    return(df)
}

我将此功能应用于DT，以创建一个新变量供列表观察计数：

DT[, count := foo(strings)]

这样做时，我会收到此错误：

Warning message:
In `[.data.table`(DT, , `:=`(counts, foo(strings))) :
  Supplied 3 items to be assigned to 4 items of column 'counts' (recycled leaving remainder of 1 items).

返回的数据表具有大小为4而不是大小3的计数变量列表，并且对于变量strings中的每个字符串观察，它均不能准确表示'A'，'AA'和'AAA'的数量。如何将列表分配为数据表中的观察值？

Answer 1

将列列表分配给单个名称可能会产生警告消息。相反，它可以是

DT[, c('n', 'n2', 'n3') := .(str_count(strings, 'A'), 
        str_count(strings, 'AA'),  str_count(strings, 'AAA'))]

Answer 2

您需要transpose list：

foo <- function(str) {
  n <- str_count(str, 'A')
  n2 <- str_count(str, 'AA')
  n3 <- str_count(str, 'AAA')
  df <- transpose(list('n' = n, 'n2' = n2, 'n3' = n3)) # <- add transpose 
  return(df)
}

DT[, count := foo(strings)]
DT
#     strings count
#  1:   AAABD 3,1,1
#  2:    BBDA 1,0,0
#  3: AACBDAA 4,2,0
#  4:   ABACD 2,0,0

str(DT)
# Classes ‘data.table’ and 'data.frame':    4 obs. of  2 variables:
# $ strings: chr  "AAABD" "BBDA" "AACBDAA" "ABACD"
# $ count  :List of 4
#  ..$ : int  3 1 1
#  ..$ : int  1 0 0
#  ..$ : int  4 2 0
#  ..$ : int  2 0 0

如何在数据表中将列表分配为观察值？

2 个答案: