2016-09-29 102 views
2

我有具有多个值的列一个数据帧列的一个热编码(逗号分隔):ř数据帧 - 包含多个术语

mydf <- structure(list(Age = c(99L, 10L, 40L, 15L), 
         Info = c("good, bad, sad", "nice, happy, joy", "NULL", "okay, nice, fun, wild, go"), 
         Target = c("Boy", "Girl", "Boy", "Boy")), 
        .Names = c("Age", "Info", "Target"), 
        row.names = c(NA, 4L), 
        class = "data.frame") 

> mydf 
    Age      Info Target 
1 99   good, bad, sad Boy 
2 10   nice, happy, joy Girl 
3 40      NULL Boy 
4 15 okay, nice, fun, wild, go Boy 

信息柱分成一热 - 编码列,并追加除了目标列的结果,例如:

Age      Info Target good bad sad nice ... NULL .. 
1 99   good, bad, sad Boy 1 1 1 0  0 
2 10   nice, happy, joy Girl 0 0 0 1  0 
3 40      NULL Boy 0 0 0 0  1 
4 15 okay, nice, fun, wild, go Boy 0 0 0 0  0 

在Python中我可以做类似下面,获得一本字典,然后用它来分配列。

In [1]: import itertools 

In [2]: values = ["good, bad, sad", "nice, happy, joy", "NULL", "okay, nice, fun, wild, go"] 

In [3]: terms = list(itertools.chain(*[v.split(", ") for v in values])) 

In [4]: dictionary = {v:k for k,v in enumerate(terms)} 

In [6]: dictionary 
Out[6]: 
{'NULL': 6, 'bad': 1, 
'fun': 9, 'go': 11, 'good': 0, 'happy': 4, 
'joy': 5, 'nice': 8, 'okay': 7, 'sad': 2, 'wild': 10} 

到目前为止,我在R中可以

> lapply(mydf["Info"], function(x) { strsplit(x, ", ") }) 
$Info 
$Info[[1]] 
[1] "good" "bad" "sad" 

$Info[[2]] 
[1] "nice" "happy" "joy" 

$Info[[3]] 
[1] "NULL" 

$Info[[4]] 
[1] "okay" "nice" "fun" "wild" "go" 

我没有得到如何将它转换成R A字典做到这一点,并用它来转换成列独热编码。

我该如何解决这个问题?

回答

5

一种选择是从mtabulate通过qdapTools,

library(qdapTools) 
cbind(mydf, mtabulate(strsplit(mydf$Info, ", "))) 
#Age      Info Target bad fun go good happy joy nice NULL okay sad wild 
#1 99   good, bad, sad Boy 1 0 0 1  0 0 0 0 0 1 0 
#2 10   nice, happy, joy Girl 0 0 0 0  1 1 1 0 0 0 0 
#3 40      NULL Boy 0 0 0 0  0 0 0 1 0 0 0 
#4 15 okay, nice, fun, wild, go Boy 0 1 1 0  0 0 1 0 1 0 1 
分裂 '信息' 栏后