R語言 實現將1對多數據與1對1數據互換
想了好長時間名字,不知道要解決的問題的名字叫什么,直接上問題demo
問題demo
現在有用戶消費金額的數據:
| 用戶 | 日期 | 金額 |
|---|---|---|
| 小明 | 2016-01 | 300 |
| 小明 | 2016-02 | 500 |
| 小明 | 2016-03 | 400 |
| 小劉 | 2016-01 | 700 |
| 小劉 | 2016-02 | 800 |
| 小劉 | 2016-03 | 600 |
我將以上數據格式為一對多數據(想不出好名字,敬請大家拍磚)
還有一種數據形式如下,我將如下格式數據稱為1對1數據
| 用戶 | 2016-01 | 2016-02 | 2016-03 |
|---|---|---|---|
| 小明 | 300 | 500 | 400 |
| 小劉 | 700 | 800 | 600 |
如何用R語言實現1對多數據與1對1數據之間的互換,在這里寫了一個簡單的小函數,大家有好的想法敬請?zhí)岢觥?/p>
生成1對1數據集的代碼如下:
#創(chuàng)建數據集
c1<-c("小明",300,500,400)
c2<-c("小劉",700,800,600)
dt<-as.data.frame(rbind(c1,c2))
names(dt)<-c("用戶","2016-01","2016-02","2016-03")
1對1數據轉成1對多數據
構建1對多數據的轉換函數:
##data原始數據集
##colList要變換的列
##要保留的主鍵列
One2More<-function(data,colList,primaryCol){
result=data.frame(NULL)
for(r in c(1:nrow(data))){
temp<-as.data.frame(t(data[r,colList]))
temp$日期<-row.names(temp)
temp<-cbind(temp,data[r,primaryCol])
names(temp)<-c("c1","c2","c3")#臨時起的名字,可按自己需求進行修改
#編行號開始,如果沒有要求,此步可省略
resultRows<-nrow(result)
tempRows<-nrow(temp)
row.names(temp)<-c((resultRows+1):(resultRows+tempRows))
#編行號結束
result<-rbind(result,temp)
}
result;
}
執(zhí)行代碼如下:
One2MoreResult<-One2More(dt,c("2016-01","2016-02","2016-03"),"用戶")
結果如下圖所示:

1對多數據轉成1對1數據
針對多轉一我們需要安裝plyr包,函數如下:
#如果有plyr,請?zhí)^此步安裝
install.packages("plyr")
library(plyr)
##data:數據集
##primaryCol:要分組的那一列
##rowNameCols:想要變成表頭的列
More2One<-function(data,primaryCol,rowNameCols){
ddply(data,primaryCol,function(k){colNames<-k[,rowNameCols];row.names(k)<-k[,rowNameCols];k<-k[,-c(which(colnames(k)==rowNameCols | colnames(k)==primaryCol ))];t(k)})
}
注意:如果使用上文中One2MoreResult數據,請注意數據類型,得出的數值結果為因子類型,請先進行轉換,轉換代碼如下:
One2MoreResult$c1<-as.numeric(as.character(One2MoreResult$c1))
此時One2MoreResult的數據如下:

此時要按照c3進行分組,將c2列放在表頭去
執(zhí)行代碼如下:
More2OneResult<-More2One(One2MoreResult,"c3","c2")
結果如下圖所示:

完整代碼
#創(chuàng)建數據集
c1<-c("小明",300,500,400)
c2<-c("小劉",700,800,600)
dt<-as.data.frame(rbind(c1,c2))
View(dt)
names(dt)<-c("用戶","2016-01","2016-02","2016-03")
##data原始數據集
##colList要變換的列
##要保留的主鍵列
One2More<-function(data,colList,primaryCol){
result=data.frame(NULL)
for(r in c(1:nrow(data))){
temp<-as.data.frame(t(data[r,colList]))
temp$日期<-row.names(temp)
temp<-cbind(temp,data[r,primaryCol])
names(temp)<-c("c1","c2","c3")
#編行號開始,如果沒有要求,此步可省略
resultRows<-nrow(result)
tempRows<-nrow(temp)
row.names(temp)<-c((resultRows+1):(resultRows+tempRows))
#編行號結束
result<-rbind(result,temp)
}
result;
}
#如果有plyr,請?zhí)^此步安裝
install.packages("plyr")
library(plyr)
##data:數據集
##primaryCol:要分組的那一列
##rowNameCols:想要變成表頭的列
More2One<-function(data,primaryCol,rowNameCols){
ddply(data,primaryCol,function(k){colNames<-k[,rowNameCols];row.names(k)<-k[,rowNameCols];k<-k[,-c(which(colnames(k)==rowNameCols | colnames(k)==primaryCol ))];t(k)})
}
One2MoreResult<-One2More(dt,c("2016-01","2016-02","2016-03"),"用戶")
View(One2MoreResult)
One2MoreResult$c1<-as.numeric(as.character(One2MoreResult$c1))
More2OneResult<-More2One(One2MoreResult,"c3","c2")
View(More2OneResult)
期望
如果有時間會解決如下問題:
1. 提高代碼的通用性
2. 列名或表頭名可以通過指定來解決
以上為個人經驗,希望能給大家一個參考,也希望大家多多支持腳本之家。如有錯誤或未考慮完全的地方,望不吝賜教。

