Подтвердить что ты не робот

Повторить строки data.frame

Я хочу повторить строки data.frame, каждый N раз. Результатом должен быть новый data.framenrow(new.df) == nrow(old.df) * N), сохраняющий типы данных столбцов.

Пример для N = 2:

                        A B   C
  A B   C             1 j i 100
1 j i 100     -->     2 j i 100
2 K P 101             3 K P 101
                      4 K P 101

Итак, каждая строка повторяется 2 раза, а символы остаются символами, факторы остаются факторами, численные значения остаются числами,...

Используется моя первая попытка: apply(old.df, 2, function(co) rep(co, each = N)), но это преобразует мои значения в символы, и я получаю:

     A   B   C    
[1,] "j" "i" "100"
[2,] "j" "i" "100"
[3,] "K" "P" "101"
[4,] "K" "P" "101"
4b9b3361

Ответ 1

df <- data.frame(a = 1:2, b = letters[1:2]) 
df[rep(seq_len(nrow(df)), each = 2), ]

Ответ 2

Чистый раствор dplyr, взятый из здесь

library(dplyr)
df <- tibble(x = 1:2, y = c("a", "b"))
df %>% slice(rep(1:n(), each = 2))

Ответ 3

Если вы можете повторить все это или подмножить сначала, а затем повторите это, тогда этот похожий вопрос может оказаться полезным. Еще раз:

library(mefa)
rep(mtcars,10) 

или просто

mefa:::rep.data.frame(mtcars)

Ответ 4

Функция rep.row, по-видимому, иногда создает списки для столбцов, что приводит к плохой памяти. Я написал следующее, которое, кажется, хорошо работает:

library(plyr)
rep.row <- function(r, n){
  colwise(function(x) rep(x, n))(r)
}

Ответ 5

Добавив к тому, что @dardisco упомянул о mefa::rep.data.frame(), он очень гибкий.

Вы можете либо повторять каждую строку N раз:

rep(df, each=N)

или повторить весь фрейм данных N раз (подумайте: например, когда вы перерабатываете векторизованный аргумент)

rep(df, times=N)

Два больших пальца вверх для mefa! Я никогда не слышал об этом до сих пор, и мне пришлось написать ручной код, чтобы сделать это.

Ответ 6

Для справки и добавления ответов, цитирующих mefa, может стоить взглянуть на реализацию mefa::rep.data.frame(), если вы не хотите включать весь пакет:

> data <- data.frame(a=letters[1:3], b=letters[4:6])
> data
  a b
1 a d
2 b e
3 c f
> as.data.frame(lapply(data, rep, 2))
  a b
1 a d
2 b e
3 c f
4 a d
5 b e
6 c f

Ответ 7

Существует прекрасное векторизованное решение, которое повторяет только определенные строки по n раз, возможно, например, добавив столбец ntimes во фрейм данных:

  A B   C ntimes
1 j i 100      2
2 K P 101      4
3 Z Z 102      1

Метод:

df <- data.frame(A=c("j","K","Z"), B=c("i","P","Z"), C=c(100,101,102), ntimes=c(2,4,1))
df <- as.data.frame(lapply(df, rep, df$ntimes))

Результат:

  A B   C ntimes
1 Z Z 102      1
2 j i 100      2
3 j i 100      2
4 K P 101      4
5 K P 101      4
6 K P 101      4
7 K P 101      4

Это очень похоже на метод Джоша О'Брайена и Марка Миллера:

df[rep(seq_len(nrow(df)), df$ntimes),]

Однако этот метод выглядит немного медленнее:

df <- data.frame(A=c("j","K","Z"), B=c("i","P","Z"), C=c(100,101,102), ntimes=c(2000,3000,4000))

microbenchmark::microbenchmark(
  df[rep(seq_len(nrow(df)), df$ntimes),],
  as.data.frame(lapply(df, rep, df$ntimes)),
  times = 10
)

Результат:

Unit: microseconds
                                      expr      min       lq      mean   median       uq      max neval
   df[rep(seq_len(nrow(df)), df$ntimes), ] 3563.113 3586.873 3683.7790 3613.702 3657.063 4326.757    10
 as.data.frame(lapply(df, rep, df$ntimes))  625.552  654.638  676.4067  668.094  681.929  799.893    10

Ответ 8

Мое решение похожее на mefa:::rep.data.frame, но немного быстрее и заботится о именах строк:

rep.data.frame <- function(x, times) {
    rnames <- attr(x, "row.names")
    x <- lapply(x, rep.int, times = times)
    class(x) <- "data.frame"
    if (!is.numeric(rnames))
        attr(x, "row.names") <- make.unique(rep.int(rnames, times))
    else
        attr(x, "row.names") <- .set_row_names(length(rnames) * times)
    x
}

Сравнить решения:

library(Lahman)
library(microbenchmark)
microbenchmark(
    mefa:::rep.data.frame(Batting, 10),
    rep.data.frame(Batting, 10),
    Batting[rep.int(seq_len(nrow(Batting)), 10), ],
    times = 10
)
#> Unit: milliseconds
#>                                            expr       min       lq     mean   median        uq       max neval cld
#>              mefa:::rep.data.frame(Batting, 10) 127.77786 135.3480 198.0240 148.1749  278.1066  356.3210    10  a 
#>                     rep.data.frame(Batting, 10)  79.70335  82.8165 134.0974  87.2587  191.1713  307.4567    10  a 
#>  Batting[rep.int(seq_len(nrow(Batting)), 10), ] 895.73750 922.7059 981.8891 956.3463 1018.2411 1127.3927    10   b

Ответ 9

попробуйте использовать, например,

N=2
rep(1:4, each = N) 

как индекс

Ответ 10

Другой способ сделать это - сначала получить индексы строк, добавить дополнительные копии df, а затем упорядочить по индексам:

df$index = 1:nrow(df)
df = rbind(df,df)
df = df[order(df$index),][,-ncol(df)]

Хотя другие решения могут быть короче, этот метод может быть более выгоден в определенных ситуациях.