重复data.frame，添加主键

Question

我有一个数据框。说，

data.frame(x = c(1, 3), y = c(5, 0), id = c("A", "B"))

现在我想复制它，所以我在同一个 data.frame 中有一个副本。我最终会得到这样的结果，

 data.frame(x = c(1, 3, 1, 3), y = c(5, 0, 5, 0), id = c("A", "B", "A", "B"))

现在，这与我想要的非常接近，但我还想附加 id 列，以根据我想要的重复数使它们对每一行都是唯一的（在这种情况下只有一个，但我想要 n 个） .

data.frame(x = c(1, 3, 1, 3), y = c(5, 0, 5, 0), id = c("A-1", "B-1", "A-2", "B-2"))

所以，如您所见，我可以全神贯注地制作对象，但我想从使用基本 R 编写“hacky”代码转变为使用 dplyr 复制此功能。

Answer 1

所以我注意到您想使用 dplyr 包执行此操作。我认为结合使用来自dplyr 的group_by()、mutate() 和row_number() 函数，您可以很好地完成这项工作。

library(dplyr)

# so you start with this data.frame:
df <- data.frame(x = c(1, 3), y = c(5, 0), id = c("A", "B"))

# to attach an exact duplication of this df to itself:
df <- rbind(df, df)


# group by id, add a second id to increment within each id group ("A", "B", etc.)
df2 <- group_by(df, id) %>%
    mutate(id2 = row_number())


# paste the id and id2 together for the desired result
df2$id_combined <- paste0(df2$id, '-', df2$id2)

# inspect results
df2
    # x     y     id   id2 id_combined
    # <dbl> <dbl> <fctr> <int>       <chr>
    # 1     1     5      A     1         A-1
    # 2     3     0      B     1         B-1
    # 3     1     5      A     2         A-2
    # 4     3     0      B     2         B-2

请记住，您现在拥有的是“tibble”/“grouped data.frame”，而不是基本的 data.frame。

如果您愿意，可以简单地将其恢复为原始 data.frame。

df2 <- data.frame(df2, stringsAsFactors = F)

# now to remove the additional columns that were added in this process:
df2$id2 <- NULL

编辑——探索将同一数据框的n复制附加到自身的其他选项：

# Not dplyr, but this is how I would normally handle this type of task:
df <- data.frame(x = c(1, 3), y = c(5, 0), id = c("A", "B"))

# set n equal to the number of times you want to replicate the data.frame
n <- 13

# initialize space to hold the data frames
list_dfs <- list()

# loop through, adding individual data frames to the list
for(i in 1:n) {
    list_dfs[[i]] <- df
}

# combine them all with do.call
my_big_df <- do.call(rbind, list_dfs)

然后，您可以使用上面显示的 group_by()、mutate() 和 row_number() 函数为 data.frame 创建新的唯一键。