首页 > 解决方案 > `tm` 包中的 `tm_map` 中的自定义函数

问题描述

我正在尝试将自定义函数添加到tm_map包的函数中,tm但是它将数据转换为我无法继续的不同格式。

例如我使用

library(tm)
library(qdapRegex)

docs <- data.frame(doc_id = c("doc_1", "doc_2"),
                   text = c("This is a text. With some more text, www.yahoo.com", "This another one. with some different text www.google.com"),
                   dmeta1 = 1:2, dmeta2 = letters[1:2],
                   stringsAsFactors = FALSE)

docs = VCorpus(DataframeSource(docs))

content(docs[[1]])

docs <- tm_map(docs, content_transformer(tolower)) # This Works fine
content(docs[[1]])

nchar_rm <- function(x){
  gsub(" *\\b[[:alpha:]]{1,2}\\b *", " ", x)
} # Custom function to remove characters less than 2

docs <- tm_map(docs, nchar_rm) # implement custom function
content(docs[[1]]) # returns an error.

错误:

Error in UseMethod("content", x) : 
  no applicable method for 'content' applied to an object of class "character"

docs <- tm_map(docs, rm_url)使用包中rm_url的函数qdapRegex也会返回错误。

标签: rtm

解决方案


content_transformer以同样的方式使用

library(tm)

docs <- tm_map(docs, content_transformer(nchar_rm)) 
content(docs[[1]])
#[1] "This  text. With some more text, www.yahoo.com"

它也将与rm_url

library(qdapRegex)

docs <- tm_map(docs, content_transformer(rm_url))
content(docs[[1]])
#[1] "This is a text. With some more text,"

但是,您可能知道您可以在完全不使用tm函数的情况下使用lapply/ sapply/map等来执行此操作。

lapply(docs$text, rm_url)
lapply(docs$text, nchar_rm)

推荐阅读