วางหลายคอลัมน์เข้าด้วยกัน


100

ฉันมีคอลัมน์จำนวนหนึ่งในดาต้าเฟรมที่ฉันต้องการวางเข้าด้วยกัน (คั่นด้วย "-") ดังนี้:

data <- data.frame('a' = 1:3, 
                   'b' = c('a','b','c'), 
                   'c' = c('d', 'e', 'f'), 
                   'd' = c('g', 'h', 'i'))
i.e.     
     a   b   c  d  
     1   a   d   g  
     2   b   e   h  
     3   c   f   i  

ที่ฉันอยากจะเป็น:

a x  
1 a-d-g  
2 b-e-h  
3 c-f-i  

โดยปกติฉันสามารถทำได้ด้วย:

within(data, x <- paste(b,c,d,sep='-'))

จากนั้นจึงลบคอลัมน์เก่าออก แต่น่าเสียดายที่ฉันไม่รู้จักชื่อของคอลัมน์โดยเฉพาะมีเพียงชื่อรวมของคอลัมน์ทั้งหมดเช่นฉันจะรู้ว่า cols <- c('b','c','d')

ไม่มีใครรู้วิธีการทำเช่นนี้?

คำตอบ:


104
# your starting data..
data <- data.frame('a' = 1:3, 'b' = c('a','b','c'), 'c' = c('d', 'e', 'f'), 'd' = c('g', 'h', 'i')) 

# columns to paste together
cols <- c( 'b' , 'c' , 'd' )

# create a new column `x` with the three columns collapsed together
data$x <- apply( data[ , cols ] , 1 , paste , collapse = "-" )

# remove the unnecessary columns
data <- data[ , !( names( data ) %in% cols ) ]

8
ไม่จำเป็นต้องสมัครที่นี่ paste เป็น vectorised และมีประสิทธิภาพมากกว่า
baptiste

1
@baptiste .. เป็นไปไม่ได้do.call?
Anthony Damico

1
แน่นอนคุณสามารถใช้ตัวอย่างเช่นevil(parse(...))แต่ฉันเชื่อว่าdo.callนี่เป็นการเรียกที่ถูกต้อง
baptiste

Do.call นี่คือเทคนิคที่ดีกว่า รักษาการกำหนดเวกเตอร์
Clayton Stanley

1
อืม .. จะผ่านไปได้ยังcollapse = "-"ไง? ถึงpaste?
Anthony Damico

48

เป็นตัวแปรในคำตอบของ baptisteโดยdataกำหนดตามที่คุณมีและคอลัมน์ที่คุณต้องการรวมเข้าด้วยกันที่กำหนดไว้ในcols

cols <- c("b", "c", "d")

คุณสามารถเพิ่มคอลัมน์ใหม่dataและลบคอลัมน์เก่าด้วย

data$x <- do.call(paste, c(data[cols], sep="-"))
for (co in cols) data[co] <- NULL

ซึ่งจะช่วยให้

> data
  a     x
1 1 a-d-g
2 2 b-e-h
3 3 c-f-i

มีเครื่องหมายจุลภาคใน "c (data [cols], ... " หรือไม่เช่นนี้: "c (data [, cols], ... "
roschu

2
@roschu ไม่ว่าจะทำงาน การสร้างดัชนีdata.frameด้วยเวกเตอร์อักขระเดียวจะเป็นการสร้างดัชนีคอลัมน์แม้ว่าอาร์กิวเมนต์แรกมักจะเป็นดัชนีแถว
Brian Diggs

รวดเร็วและชาญฉลาด ขอบคุณ
Ali Khosro

33

การใช้tidyrแพคเกจนี้สามารถจัดการได้อย่างง่ายดายในการเรียกใช้ฟังก์ชัน 1 ครั้ง

data <- data.frame('a' = 1:3, 
                   'b' = c('a','b','c'), 
                   'c' = c('d', 'e', 'f'), 
                   'd' = c('g', 'h', 'i'))

tidyr::unite_(data, paste(colnames(data)[-1], collapse="_"), colnames(data)[-1])

  a b_c_d
1 1 a_d_g
2 2 b_e_h
3 3 c_f_i

แก้ไข:ยกเว้นคอลัมน์แรกทุกอย่างจะถูกวาง

# tidyr_0.6.3

unite(data, newCol, -a) 
# or by column index unite(data, newCol, -1)

#   a newCol
# 1 1  a_d_g
# 2 2  b_e_h
# 3 3  c_f_i

3
ฉันคิดว่า OP พูดถึงพวกเขาไม่ทราบชื่อคอลัมน์ล่วงหน้ามิฉะนั้นอาจทำได้within(data, x <- paste(b,c,d,sep='-'))ตามที่แสดงไว้
David Arenburg

ฉันเห็นด้วยกับ @DavidArenburg สิ่งนี้ไม่ได้กล่าวถึงสถานการณ์ของ OP ฉันคิดว่าunite_(data, "b_c_d", cols)จะหรือขึ้นอยู่กับ data.frame จริงของพวกเขาunite(data, b_c_d, -a)อาจเป็นตัวเลือกเช่นกัน
Sam Firke

14

ฉันจะสร้าง data.frame ใหม่:

d <- data.frame('a' = 1:3, 'b' = c('a','b','c'), 'c' = c('d', 'e', 'f'), 'd' = c('g', 'h', 'i')) 

cols <- c( 'b' , 'c' , 'd' )

data.frame(a = d[, 'a'], x = do.call(paste, c(d[ , cols], list(sep = '-'))))

โปรดทราบว่าd[ , cols]คุณอาจต้องการใช้แทนd[ , names(d) != 'a']หากaต้องวางคอลัมน์ทั้งหมดไว้ด้วยกัน
baptiste

2
หนึ่งในวิธีแก้ปัญหาที่เป็นที่ยอมรับใน SO ฉันคิดว่าคุณสามารถย่อสิ่งนี้ให้สั้นลงcbind(a = d['a'], x = do.call(paste, c(d[cols], sep = '-')))เช่นหลีกเลี่ยงเครื่องหมายจุลภาคlistและdata.frameในขณะที่ใช้data.frameวิธีการcbind
David Arenburg

9

เพียงเพื่อเพิ่มโซลูชันเพิ่มเติมReduceซึ่งอาจจะช้ากว่าdo.callแต่ดีกว่าการตรวจสอบมากกว่าapplyเพราะจะหลีกเลี่ยงการmatrixแปลง นอกจากนี้แทนที่จะเป็นforลูปที่เราสามารถใช้setdiffเพื่อลบคอลัมน์ที่ไม่ต้องการได้

cols <- c('b','c','d')
data$x <- Reduce(function(...) paste(..., sep = "-"), data[cols])
data[setdiff(names(data), cols)]
#   a     x
# 1 1 a-d-g
# 2 2 b-e-h
# 3 3 c-f-i

อีกทางเลือกหนึ่งที่เราสามารถอัปเดตได้dataโดยใช้data.tableแพ็คเกจ (สมมติว่าเป็นข้อมูลใหม่)

library(data.table)
setDT(data)[, x := Reduce(function(...) paste(..., sep = "-"), .SD[, mget(cols)])]
data[, (cols) := NULL]
data
#    a     x
# 1: 1 a-d-g
# 2: 2 b-e-h
# 3: 3 c-f-i

อีกทางเลือกหนึ่งคือใช้.SDcolsแทนmgetใน

setDT(data)[, x := Reduce(function(...) paste(..., sep = "-"), .SD), .SDcols = cols]

5

ฉันเปรียบเทียบคำตอบของ Anthony Damico, Brian Diggs และ data_steve กับตัวอย่างขนาดเล็กtbl_dfและได้ผลลัพธ์ต่อไปนี้

> data <- data.frame('a' = 1:3, 
+                    'b' = c('a','b','c'), 
+                    'c' = c('d', 'e', 'f'), 
+                    'd' = c('g', 'h', 'i'))
> data <- tbl_df(data)
> cols <- c("b", "c", "d")
> microbenchmark(
+     do.call(paste, c(data[cols], sep="-")),
+     apply( data[ , cols ] , 1 , paste , collapse = "-" ),
+     tidyr::unite_(data, "x", cols, sep="-")$x,
+     times=1000
+ )
Unit: microseconds
                                         expr     min      lq      mean  median       uq       max neval
do.call(paste, c(data[cols], sep = "-"))       65.248  78.380  93.90888  86.177  99.3090   436.220  1000
apply(data[, cols], 1, paste, collapse = "-") 223.239 263.044 313.11977 289.514 338.5520   743.583  1000
tidyr::unite_(data, "x", cols, sep = "-")$x   376.716 448.120 556.65424 501.877 606.9315 11537.846  1000

อย่างไรก็ตามเมื่อฉันประเมินtbl_dfด้วยตัวเองด้วย ~ 1 ล้านแถวและ 10 คอลัมน์ผลลัพธ์ก็แตกต่างกันมาก

> microbenchmark(
+     do.call(paste, c(data[c("a", "b")], sep="-")),
+     apply( data[ , c("a", "b") ] , 1 , paste , collapse = "-" ),
+     tidyr::unite_(data, "c", c("a", "b"), sep="-")$c,
+     times=25
+ )
Unit: milliseconds
                                                       expr        min         lq      mean     median        uq       max neval
do.call(paste, c(data[c("a", "b")], sep="-"))                 930.7208   951.3048  1129.334   997.2744  1066.084  2169.147    25
apply( data[ , c("a", "b") ] , 1 , paste , collapse = "-" )  9368.2800 10948.0124 11678.393 11136.3756 11878.308 17587.617    25
tidyr::unite_(data, "c", c("a", "b"), sep="-")$c              968.5861  1008.4716  1095.886  1035.8348  1082.726  1759.349    25

5

ในความคิดของฉันsprintfฟังก์ชั่นนี้สมควรได้รับในคำตอบเหล่านี้ คุณสามารถใช้sprintfดังนี้:

do.call(sprintf, c(d[cols], '%s-%s-%s'))

ซึ่งจะช่วยให้:

 [1] "a-d-g" "b-e-h" "c-f-i"

และเพื่อสร้าง dataframe ที่ต้องการ:

data.frame(a = d$a, x = do.call(sprintf, c(d[cols], '%s-%s-%s')))

การให้:

  a     x
1 1 a-d-g
2 2 b-e-h
3 3 c-f-i

แม้ว่าจะsprintfไม่ได้มีข้อได้เปรียบที่ชัดเจนกว่าdo.call/ pasteการรวมกันของ @BrianDiggs มันเป็นโดยเฉพาะอย่างยิ่งมีประโยชน์เมื่อคุณยังต้องการที่จะแผ่นบางส่วนของสตริงที่ต้องการหรือเมื่อคุณต้องการที่จะระบุจำนวนหลัก ดู?sprintfตัวเลือกต่างๆ

ตัวแปรอื่นจะใช้pmapจาก:

pmap(d[2:4], paste, sep = '-')

หมายเหตุ: pmapโซลูชันนี้ใช้ได้เฉพาะเมื่อคอลัมน์ไม่ใช่ปัจจัย


เกณฑ์มาตรฐานของชุดข้อมูลขนาดใหญ่:

# create a larger dataset
d2 <- d[sample(1:3,1e6,TRUE),]
# benchmark
library(microbenchmark)
microbenchmark(
  docp = do.call(paste, c(d2[cols], sep="-")),
  appl = apply( d2[, cols ] , 1 , paste , collapse = "-" ),
  tidr = tidyr::unite_(d2, "x", cols, sep="-")$x,
  docs = do.call(sprintf, c(d2[cols], '%s-%s-%s')),
  times=10)

ผลลัพธ์ใน:

Unit: milliseconds
 expr       min        lq      mean    median        uq       max neval cld
 docp  214.1786  226.2835  297.1487  241.6150  409.2495  493.5036    10 a  
 appl 3832.3252 4048.9320 4131.6906 4072.4235 4255.1347 4486.9787    10   c
 tidr  206.9326  216.8619  275.4556  252.1381  318.4249  407.9816    10 a  
 docs  413.9073  443.1550  490.6520  453.1635  530.1318  659.8400    10  b 

ข้อมูลที่ใช้:

d <- data.frame(a = 1:3, b = c('a','b','c'), c = c('d','e','f'), d = c('g','h','i')) 

3

นี่เป็นวิธีการที่ค่อนข้างแปลก (แต่รวดเร็ว): ใช้fwriteจากdata.tableเพื่อ "วาง" คอลัมน์เข้าด้วยกันและfreadอ่านย้อนกลับเพื่อความสะดวกฉันได้เขียนขั้นตอนเป็นฟังก์ชันที่เรียกว่าfpaste:

fpaste <- function(dt, sep = ",") {
  x <- tempfile()
  fwrite(dt, file = x, sep = sep, col.names = FALSE)
  fread(x, sep = "\n", header = FALSE)
}

นี่คือตัวอย่าง:

d <- data.frame(a = 1:3, b = c('a','b','c'), c = c('d','e','f'), d = c('g','h','i')) 
cols = c("b", "c", "d")

fpaste(d[cols], "-")
#       V1
# 1: a-d-g
# 2: b-e-h
# 3: c-f-i

มันทำงานอย่างไร?

d2 <- d[sample(1:3,1e6,TRUE),]
  
library(microbenchmark)
microbenchmark(
  docp = do.call(paste, c(d2[cols], sep="-")),
  tidr = tidyr::unite_(d2, "x", cols, sep="-")$x,
  docs = do.call(sprintf, c(d2[cols], '%s-%s-%s')),
  appl = apply( d2[, cols ] , 1 , paste , collapse = "-" ),
  fpaste = fpaste(d2[cols], "-")$V1,
  dt2 = as.data.table(d2)[, x := Reduce(function(...) paste(..., sep = "-"), .SD), .SDcols = cols][],
  times=10)
# Unit: milliseconds
#    expr        min         lq      mean     median         uq       max neval
#    docp  215.34536  217.22102  220.3603  221.44104  223.27224  225.0906    10
#    tidr  215.19907  215.81210  220.7131  220.09636  225.32717  229.6822    10
#    docs  281.16679  285.49786  289.4514  286.68738  290.17249  312.5484    10
#    appl 2816.61899 3106.19944 3259.3924 3266.45186 3401.80291 3804.7263    10
#  fpaste   88.57108   89.67795  101.1524   90.59217   91.76415  197.1555    10
#     dt2  301.95508  310.79082  384.8247  316.29807  383.94993  874.4472    10

ถ้าคุณเขียนและอ่านลง ramdisk ล่ะ? การเปรียบเทียบจะยุติธรรมกว่าเล็กน้อย
jangorecki

@jangorecki ไม่แน่ใจว่าฉันทำถูกต้องหรือเปล่า (ฉันเริ่ม R ด้วยTMPDIR=/dev/shm R) แต่ฉันไม่สังเกตเห็นความแตกต่างอย่างมากเมื่อเทียบกับผลลัพธ์เหล่านี้ ฉันยังไม่ได้เล่นเลยกับจำนวนเธรดที่ใช้freadหรือfwriteเพื่อดูว่ามันมีผลต่อผลลัพธ์อย่างไร
A5C1D2H2I1M1N2O1R2T1

1
library(plyr)

ldply(apply(data, 1, function(x) data.frame(
                      x = paste(x[2:4],sep="",collapse="-"))))

#      x
#1 a-d-g
#2 b-e-h
#3 c-f-i

#  and with just the vector of names you have:

ldply(apply(data, 1, function(x) data.frame(
                      x = paste(x[c('b','c','d')],sep="",collapse="-"))))

# or equally:
mynames <-c('b','c','d')
ldply(apply(data, 1, function(x) data.frame(
                      x = paste(x[mynames],sep="",collapse="-"))))    

0

ฉันรู้ว่านี่เป็นคำถามเก่า แต่คิดว่าฉันควรจะนำเสนอวิธีง่ายๆโดยใช้ฟังก์ชัน paste () ตามที่ผู้ถามแนะนำ:

data_1<-data.frame(a=data$a,"x"=paste(data$b,data$c,data$d,sep="-")) 
data_1
  a     x
1 1 a-d-g
2 2 b-e-h
3 3 c-f-i
โดยการใช้ไซต์ของเรา หมายความว่าคุณได้อ่านและทำความเข้าใจนโยบายคุกกี้และนโยบายความเป็นส่วนตัวของเราแล้ว
Licensed under cc by-sa 3.0 with attribution required.