Додати новий рядок до фрейму даних у певному індексі рядка, не доданий?


160

Наступний код поєднує вектор з фреймом даних:

newrow = c(1:4)
existingDF = rbind(existingDF,newrow)

Однак цей код завжди вставляє новий рядок в кінці фрейму даних.

Як я можу вставити рядок у вказаній точці в рамці даних? Наприклад, скажімо, що у фрейму даних є 20 рядків, як я можу вставити новий рядок між рядками 10 та 11?


Використовувати зручний індекс та сортувати?
Роланд

22
existingDF = rbind(existingDF[1:10,],newrow,existingDF[-(1:10),])
Поп

За допомогою простого циклу та умови, якщо потрібно, рядки можна додавати з одного фрейму даних в інший. Зразок коду, як показано нижчеnewdataframe[nrow(newdataframe)+1,] <- existingdataframe[i,]
kirancodify

Відповіді:


156

Ось рішення, яке дозволяє уникнути (часто повільного) rbindдзвінка:

existingDF <- as.data.frame(matrix(seq(20),nrow=5,ncol=4))
r <- 3
newrow <- seq(4)
insertRow <- function(existingDF, newrow, r) {
  existingDF[seq(r+1,nrow(existingDF)+1),] <- existingDF[seq(r,nrow(existingDF)),]
  existingDF[r,] <- newrow
  existingDF
}

> insertRow(existingDF, newrow, r)
  V1 V2 V3 V4
1  1  6 11 16
2  2  7 12 17
3  1  2  3  4
4  3  8 13 18
5  4  9 14 19
6  5 10 15 20

Якщо швидкість менш важлива, ніж ясність, то рішення @ Саймона працює добре:

existingDF <- rbind(existingDF[1:r,],newrow,existingDF[-(1:r),])
> existingDF
   V1 V2 V3 V4
1   1  6 11 16
2   2  7 12 17
3   3  8 13 18
4   1  2  3  4
41  4  9 14 19
5   5 10 15 20

(Зверніть увагу, ми індексуємо по- rрізному).

І нарешті, орієнтири:

library(microbenchmark)
microbenchmark(
  rbind(existingDF[1:r,],newrow,existingDF[-(1:r),]),
  insertRow(existingDF,newrow,r)
)

Unit: microseconds
                                                    expr     min       lq   median       uq       max
1                       insertRow(existingDF, newrow, r) 660.131 678.3675 695.5515 725.2775   928.299
2 rbind(existingDF[1:r, ], newrow, existingDF[-(1:r), ]) 801.161 831.7730 854.6320 881.6560 10641.417

Орієнтири

Як на мене завжди вказує @MatthewDowle, орієнтири потрібно вивчити на предмет масштабування, оскільки розмір проблеми збільшується. Ось ми тоді:

benchmarkInsertionSolutions <- function(nrow=5,ncol=4) {
  existingDF <- as.data.frame(matrix(seq(nrow*ncol),nrow=nrow,ncol=ncol))
  r <- 3 # Row to insert into
  newrow <- seq(ncol)
  m <- microbenchmark(
   rbind(existingDF[1:r,],newrow,existingDF[-(1:r),]),
   insertRow(existingDF,newrow,r),
   insertRow2(existingDF,newrow,r)
  )
  # Now return the median times
  mediansBy <- by(m$time,m$expr, FUN=median)
  res <- as.numeric(mediansBy)
  names(res) <- names(mediansBy)
  res
}
nrows <- 5*10^(0:5)
benchmarks <- sapply(nrows,benchmarkInsertionSolutions)
colnames(benchmarks) <- as.character(nrows)
ggplot( melt(benchmarks), aes(x=Var2,y=value,colour=Var1) ) + geom_line() + scale_x_log10() + scale_y_log10()

@ Рішення Роланда досить добре масштабується, навіть із закликом rbind:

                                                              5       50     500    5000    50000     5e+05
insertRow2(existingDF, newrow, r)                      549861.5 579579.0  789452 2512926 46994560 414790214
insertRow(existingDF, newrow, r)                       895401.0 905318.5 1168201 2603926 39765358 392904851
rbind(existingDF[1:r, ], newrow, existingDF[-(1:r), ]) 787218.0 814979.0 1263886 5591880 63351247 829650894

Нанесено у лінійному масштабі:

лінійний

І масштаб журналу журналу:

log-log


3
Вставлення рядка в кінці викликає дивну поведінку!
Маартен

@Maarten За допомогою якої функції?
Арі Б. Фрідман

Я припускаю , що це те ж саме дивна поведінка , що я описую тут: stackoverflow.com/questions/19927806 / ...
PatrickT

1
Дивна поведінка не відбувається з insertRow2, в моєму конкретному кадрі даних та рядку.
PatrickT

Як ви просто додаєте рядок чисел до df? У мене є dfстовпці, a,b,c,dі я хочу додати рядок 1,2,3,4. Як це зробити?
Тревіс Хетер

44
insertRow2 <- function(existingDF, newrow, r) {
  existingDF <- rbind(existingDF,newrow)
  existingDF <- existingDF[order(c(1:(nrow(existingDF)-1),r-0.5)),]
  row.names(existingDF) <- 1:nrow(existingDF)
  return(existingDF)  
}

insertRow2(existingDF,newrow,r)

  V1 V2 V3 V4
1  1  6 11 16
2  2  7 12 17
3  1  2  3  4
4  3  8 13 18
5  4  9 14 19
6  5 10 15 20

microbenchmark(
+   rbind(existingDF[1:r,],newrow,existingDF[-(1:r),]),
+   insertRow(existingDF,newrow,r),
+   insertRow2(existingDF,newrow,r)
+ )
Unit: microseconds
                                                    expr     min       lq   median       uq      max
1                       insertRow(existingDF, newrow, r) 513.157 525.6730 531.8715 544.4575 1409.553
2                      insertRow2(existingDF, newrow, r) 430.664 443.9010 450.0570 461.3415  499.988
3 rbind(existingDF[1:r, ], newrow, existingDF[-(1:r), ]) 606.822 625.2485 633.3710 653.1500 1489.216

3
Це класне рішення. Досі не можу зрозуміти, чому це набагато швидше, ніж одночасний дзвінок rbind, але я заінтригований.
Арі Б. Фрідман

Відповіді з орієнтирами повинні мати деяку додаткову репутацію, що застосовується автоматично ІМО. Дякую!
Алекс

10

Спробуйте спробувати пакет dplyr

library(dplyr)
a <- data.frame(A = c(1, 2, 3, 4),
               B = c(11, 12, 13, 14))


system.time({
for (i in 50:1000) {
    b <- data.frame(A = i, B = i * i)
    a <- bind_rows(a, b)
}

})

Вихідні дані

   user  system elapsed 
   0.25    0.00    0.25

На відміну від використання функції rbind

a <- data.frame(A = c(1, 2, 3, 4),
                B = c(11, 12, 13, 14))


system.time({
    for (i in 50:1000) {
        b <- data.frame(A = i, B = i * i)
        a <- rbind(a, b)
    }

})

Вихідні дані

   user  system elapsed 
   0.49    0.00    0.49 

Є певний приріст продуктивності.


-4

наприклад, ви хочете додати рядки змінної 2 до змінної 1 даних з назвою "краї", просто зробіть це так

allEdges <- data.frame(c(edges$V1,edges$V2))
Використовуючи наш веб-сайт, ви визнаєте, що прочитали та зрозуміли наші Політику щодо файлів cookie та Політику конфіденційності.
Licensed under cc by-sa 3.0 with attribution required.