仅保留 RSQLite 列中的第一个单词

Question

我正在将 SQL 代码转移到 R。

我有一个数据集，想从新列的列中提取第一个单词。

数据集

 df

        Col_A    ColB
  ABC JEn JDn       9
  nemmdn mend      10
pwek ne me ne      12

预期输出：

        Col_A    ColB    Col_A_Abbreviation
  ABC JEn JDn       9                   ABC
  nemmdn mend      10                nemmdn
pwek ne me ne      12                  pwek

以下是我目前拥有的：

library(DBI)
library(RSQLite)
mydb <- dbConnect(RSQLite::SQLite(), "")
dbWriteTable(mydb, "df_1", df)

dbGetQuery(mydb, 
'SELECT IF(INSTRB(Col_A, \' \', 1, 1) >0, 
SUBSTR(Col_A, 1, INSTRB(Col_A, \' \', 1, 1) - 1), 
Col_A) AS Col_A_Abbreviation FROM df_1')

这是我的错误信息：

Error in result_create(conn@ptr, statement) : no such function: IF

RSQLite 对它可以使用的功能有限制吗？

Answer 1

1) sqldf/RSQLite 如果您使用的是 SQLite 后端，那么您只能使用 RSQLite 支持的函数。试试这个，df 在最后的注释中被重复定义。

library(sqldf)
sqldf("select 
    *, 
    substr([Col_A], 1, instr([Col_A], ' ') - 1) FirstWord 
  from df")

给予：

          Col_A ColB FirstWord
1   ABC JEn JDn    9       ABC
2   nemmdn mend   10    nemmdn
3 pwek ne me ne   12      pwek

1a) 如果条目只能包含一个单词，则在末尾附加一个空格以确保始终至少有一个空格可以匹配。

sqldf("select 
    *, 
    substr([Col_A], 1, instr([Col_A] || ' ', ' ')-1) FirstWord 
  from df")

2) 基础R

在基础 R 中，删除第一个空格及其后面的所有内容：

transform(df, FirstWord = sub(" .*", "", Col_A))

注意

df <-
structure(list(`Col_A` = c("ABC JEn JDn", "nemmdn mend", "pwek ne me ne"
), `ColB` = c(9L, 10L, 12L)), class = "data.frame", row.names = c(NA, 
-3L))

Answer 2

除了 Clemsang 的评论，为什么在从数据集中得到结果后不使用strsplit()？

类似这样的：

a <- c("ABC Jen JDn", "nemmdn mend", "pwek ne me ne")
b <- sapply(strsplit(a, " "), "[", 1)