模糊匹配电影标题而不使用循环并按发布日期提取等效标题

时间:2015-01-29 15:59:56

标签: r matching string-matching levenshtein-distance fuzzy-search

我正在尝试使用模糊字符串匹配合并基于包含电影名称的电影标题列的两个数据集。下面给出了来自2个数据集的样本。

第一个数据集看起来像

  itemid userid rating       time                              title release_date
99995    1677    854      3 1997-12-22                      sweet nothing         1995
99996    1678    863      1 1998-03-07                         mat' i syn         1997
99997    1679    863      3 1998-03-07                          b. monkey         1998
99998    1429    863      2 1998-03-07                      sliding doors         1998
99999    1681    896      3 1998-02-11                       you so crazy         1994
100000   1682    916      3 1997-11-29 scream of stone (schrei aus stein)         1991

第二个是

itemid userid rating       time                     title release_date
117201 3175936   9140      3 2013-09-22 bei tou zou de na wu nian         2013
117202 3175936  17439      3 2013-09-18 bei tou zou de na wu nian         2013
117203 3181128   3024      5 2013-09-13                mac & jack         2013
117204 3181962  17310      5 2013-09-19         the last shepherd         2013
117205 3188690  13551      5 2013-09-17     the making of a queen         2013
117206 3198468   5338      3 2013-09-22          north 24 kaatham         2013

dput - df1

structure(list(itemid = c(1677L, 1678L, 1679L, 1429L, 1681L, 
1682L), userid = c(854L, 863L, 863L, 863L, 896L, 916L), rating = c(3L, 
1L, 3L, 2L, 3L, 3L), time = structure(c(10217, 10292, 10292, 
10292, 10268, 10194), class = "Date"), title = c("sweet nothing", 
"mat' i syn", "b. monkey", "sliding doors", "you so crazy", "scream of stone (schrei aus stein)"
), release_date = c("1995", "1997", "1998", "1998", "1994", "1991"
)), .Names = c("itemid", "userid", "rating", "time", "title", 
"release_date"), row.names = 99995:100000, class = "data.frame")

dput - df2

structure(list(itemid = c(3175936L, 3175936L, 3181128L, 3181962L, 
3188690L, 3198468L), userid = c(9140L, 17439L, 3024L, 17310L, 
13551L, 5338L), rating = c(3, 3, 5, 5, 5, 3), time = structure(c(15970, 
15966, 15961, 15967, 15965, 15970), class = "Date"), title = c("bei tou zou de na wu nian", 
"bei tou zou de na wu nian", "mac & jack", "the last shepherd", 
"the making of a queen", "north 24 kaatham"), release_date = c("2013", 
"2013", "2013", "2013", "2013", "2013")), .Names = c("itemid", 
"userid", "rating", "time", "title", "release_date"), row.names = 117201:117206, class = "data.frame")
what I am looking forward to do is fuz

zy使用(levenshteinSim)匹配两个数据集中的标题,并且对于相似度大于0.85的标题,例如,将两个数据集中的该电影的信息提取到新数据集中。同时我需要检查匹配的标题是否具有相同的发布日期,具有完全相同名称的电影可以具有多个发布日期。

有人可以指导我如何完成这项任务吗?

到目前为止,我已尝试过以下代码:

df <- sapply(df1$title,lenvenshteinSim,df2$title)

这给出了一个尺寸为11451 X 1682的矩阵。其中每列是来自第一个数据帧的单个电影标题,而行包含相似度值。我可以在这里放一个循环,或者可以看看熔化&amp; dcast以max(相似)&gt; 0.85拉出列但这看起来不是一种有效的方式。另外,我无法匹配此代码中的发布日期。

任何和所有帮助将不胜感激。

感谢。

1 个答案:

答案 0 :(得分:2)

您可以合并这些数据框

z <- merge(df1,df2,by='release_date',suffixes=c('.df1','.df2'))

会为您提供笛卡尔积(即df1df2之间所有可能的相同release_date组合,然后通过以下方式计算Levenshtein距离:

z$L.dist <- lenvenshteinSim(z$title.df1,z$title.df2)

拥有z$L.dist,您可以过滤所需的行:

subset(z,L.dist > 0.85)

<强>更新

以下是使用data.table的类似方法,这可能是更快的替代方法:

library(data.table)
d1 <- as.data.table(df1)
d2 <- as.data.table(df2)
setkey(d1,release_date)
setkey(d2,release_date)

z <- d1[d2,allow.cartesian=T,nomatch=F]

#z[,L.dist:=lenvenshteinSim(title,i.title)]
z[,L.dist:=mapply(lenvenshteinSim,title,i.title)]


z[L.dist > 0.8]