Оставьте только первое слово в колонке RSQLite - PullRequest
0 голосов
/ 07 марта 2019

Я работаю над переносом кода 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 ограничения на то, какие функции он может использовать?

Ответы [ 2 ]

1 голос
/ 07 марта 2019

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))
0 голосов
/ 07 марта 2019

Помимо комментария от Клемсанга, почему бы не использовать strsplit() после получения результата из набора данных?

Примерно так:

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