我正在尝试使用模糊字符串匹配合并基于包含电影名称的电影标题列的两个数据集。下面给出了来自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拉出列但这看起来不是一种有效的方式。另外,我无法匹配此代码中的发布日期。
任何和所有帮助将不胜感激。
感谢。
答案 0 :(得分:2)
您可以合并这些数据框
z <- merge(df1,df2,by='release_date',suffixes=c('.df1','.df2'))
会为您提供笛卡尔积(即df1
和df2
之间所有可能的相同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]