Julia：如何在GroupedDataFrame中添加或移除分组键？

我有一个DataFrame，需要基于不同分组列执行操作：对按A、B分组的DataFrame执行操作x，对仅按B分组的DataFrame执行操作y，是否需要两次分组？

案例1：从[:A, :B]分组中移除A，仅保留B分组 #

初始代码：

df=DataFrame(rand(160,3), :auto)
rename!(df,[:A,:B,:Z])

@. df.B = ifelse(rand() < 0.5, 1, 2)
@. df.A = ifelse(rand() < 0.5, 1, 2)

# 按A和B分组
gd = groupby(df, [:A, :B])

# 执行按A、B分组的操作...
# 现在需要仅按B分组执行操作

你尝试的gd.removegroup([:A])、gd.removekey([:A])、gd.ungroup([:A])都是错误的，DataFrames.jl的GroupedDataFrame没有这些方法。

正确做法 #

直接基于原DataFrame重新分组，或者通过分组对象的父DataFrame生成新分组：

# 方法1：直接用原DataFrame重新分组
gd_b = groupby(df, :B)

# 方法2：从已有的分组对象获取父DataFrame再分组
gd_b = groupby(parent(gd), :B)

两种方式本质都是基于原数据重新计算分组，性能差异可忽略。

案例2：从[:B]分组中添加A，变成[:B, :A]分组 #

初始代码：

df=DataFrame(rand(160,3), :auto)
rename!(df,[:A,:B,:Z])

@. df.B = ifelse(rand() < 0.5, 1, 2)
@. df.A = ifelse(rand() < 0.5, 1, 2)

# 按B分组
gd = groupby(df, [:B])

# 执行按B分组的操作...
# 现在需要按B和A分组执行操作

你尝试的groupby(gd, [:A])会生成嵌套分组（NestedGroupedDataFrame），不是平级的[:B,:A]分组；gd.addkey([:A])、gd.addgroup([:A])也都是不存在的方法。

正确做法 #

直接基于原DataFrame生成新的分组：

# 方法1：直接指定完整的分组键（得到平级联合分组）
gd_ba = groupby(df, [:B, :A])

# 方法2：在现有分组基础上细分（得到嵌套分组，适合分层处理）
nested_gd = groupby(gd, :A)

如果需要每个组对应唯一的B和A组合，优先用方法1；如果需要在B组内再按A细分处理，再用嵌套分组方式。

内容的提问来源于stack exchange，提问作者edjmir