r - data.frame 按列分组

Question

这个问题已经有答案了:

How to sum a variable by group (18 个回答)

已关闭 6 年前。

我有一个数据框 DF。

假设 DF 是:

  A B
1 1 2
2 1 3
3 2 3
4 3 5
5 3 6 

现在我想将 A 列的行组合在一起，并获得 B 列的总和。

例如:

  A B
1 1 5
2 2 3
3 3 11

我目前正在使用带有 sqldf 函数的 SQL 查询来执行此操作。但由于某种原因，它非常慢。有没有更方便的方法来做到这一点？我也可以使用 for 循环手动执行此操作，但它又很慢。我的 SQL 查询是“从 A 的 DF 组中选择 A,Count(B)”。

一般来说，每当我不使用向量化操作而使用 for 循环时，即使对于单个过程，性能也非常慢。

Answer 1

这是一个常见问题。在基础中，您要寻找的选项是aggregate。假设您的 data.frame 名为“mydf”，您可以使用以下内容。

> aggregate(B ~ A, mydf, sum)
  A  B
1 1  5
2 2  3
3 3 11

我还建议查看“data.table”包。

> library(data.table)
> DT <- data.table(mydf)
> DT[, sum(B), by = A]
   A V1
1: 1  5
2: 2  3
3: 3 11