是否仅选择每个h2o dataframe group_by组中的前几行(用于合并)？

import h2o

# 假设有一个名为df的h2o dataframe，包含要分组的列group和其他列data
df = ...

# 对数据进行分组
grouped = df.group_by("group")

# 选择每个组中的前几行进行合并
merged = None
num_rows_to_select = 5  # 选择每个组中的前5行
for group_value in grouped.unique("group").as_data_frame()["group"]:
    group_df = grouped.get_group(group_value)
    selected_rows = group_df.slice(0, num_rows_to_select-1)
    if merged is None:
        merged = selected_rows
    else:
        merged = merged.rbind(selected_rows)

# 输出合并后的结果
print(merged)