r - 按组将唯一/不同值的计数添加到原始数据

Question

我希望通过对第二个变量进行分组来计算唯一值的数量，然后将计数作为新列添加到现有的 data.frame 中。例如，如果现有数据框如下所示:

  color  type
1 black chair
2 black chair
3 black  sofa
4 green  sofa
5 green  sofa
6   red  sofa
7   red plate
8  blue  sofa
9  blue plate
10 blue chair

我想为每种颜色添加数据中存在的唯一类型的计数:

  color  type unique_types
1 black chair            2
2 black chair            2
3 black  sofa            2
4 green  sofa            1
5 green  sofa            1
6   red  sofa            2
7   red plate            2
8  blue  sofa            3
9  blue plate            3
10 blue chair            3

我希望使用ave，但似乎找不到不需要很多行的简单方法。我有 >100,000 行，所以我也不确定效率有多重要。

与此问题有些相似:Count number of observations/rows per group and add result to data frame

Answer 1

这是一个带有 dplyr 的解决方案包 - 它有 n_distinct() 作为 length(unique()) 的包装器。

df %>%
  group_by(color) %>%
  mutate(unique_types = n_distinct(type))