r - 创建一个具有不同列的元素计数的列

Question

我有一个DF

 x y z
 1 1 1
 1 2 5
 2 2 5
 3 3 5

我想在 R 中使用一个简单的命令来生成一个新的 DF 列，它是我原始 DF 的计数，如下所示:

 x y z xcount ycount zcount
 1 1 1 2      1      1
 1 2 5 2      2      3
 2 2 5 1      2      3
 3 3 5 1      1      3

我知道这是一个简单的问题，也许我思考这个问题的方式也受到我之前使用 Stata 的经验的限制。我已经研究过使用 table 函数，但该函数返回的向量的长度等于唯一值的数量。我可以将这些值映射到 R 中的新列，但这看起来很笨拙。

Answer 1

这是一长串:

cbind(df, setNames(lapply(df, function(x) ave(x, x, FUN = length)),
  paste0(names(df), "count")))
#  x y z xcount ycount zcount
#1 1 1 1      2      1      1
#2 1 2 5      2      2      3
#3 2 2 5      1      2      3
#4 3 3 5      1      1      3

重要的部分是:

> lapply(df, function(x) ave(x, x, FUN = length))
$x
[1] 2 2 1 1

$y
[1] 1 2 2 1

$z
[1] 1 3 3 3

剩下的就是化妆品。

---

您可以将其放入一个函数中以便于处理:

dfcount <- function(df) cbind(df, setNames(lapply(df, function(x) ave(x, x, FUN = length)), paste0(names(df), "count")))
dfcount(df)
#  x y z xcount ycount zcount
#1 1 1 1      2      1      1
#2 1 2 5      2      2      3
#3 2 2 5      1      2      3
#4 3 3 5      1      1      3