在R语言中为数据框按每3行分组添加列的技术问询

嘿，这需求很常见，不管你用R还是Python，都有超简洁的实现方式，我给你整理两种常用工具的方案：

在R中实现 #

这里有两种直观的写法，任选其一就行：

• 用专门生成分组标识的gl()函数，一步到位：

# 假设你的数据框名为df
df$Y <- as.integer(gl(n = 3, k = 3, length = nrow(df)))

解释：n是总组数，k是每组的行数，length对应数据框的总行数，转成整数后就得到1、1、1、2、2、2、3、3、3的分组标识。

• 用整数除法结合ceiling()向上取整，逻辑更易懂：

df$Y <- ceiling(seq(nrow(df)) / 3)

解释：生成从1到9的序列，每个数除以3后向上取整，正好对应每3行一组的分组号。

在Python（Pandas）中实现 #

用Pandas处理的话，这几种方法都能快速搞定：

• 用numpy.repeat()直接重复分组号：

import pandas as pd
import numpy as np

# 假设你的数据框名为df
df['Y'] = np.repeat([1, 2, 3], repeats=3)

解释：把1、2、3各自重复3次，刚好匹配9行数据的分组需求。

• 利用索引做整数除法，灵活适配任意行数：

df['Y'] = (df.index // 3) + 1

解释：默认索引从0开始，0-2除以3得0，加1后是1；3-5除以3得1，加1后是2，以此类推，哪怕数据行数不是3的倍数，这个方法也能自动把最后不足一组的归为最后一个分组。

• 用pd.cut()分割索引区间：

df['Y'] = pd.cut(df.index, bins=3, labels=[1,2,3], include_lowest=True).astype(int)

解释：把索引分成3个等距区间，分别标记为1、2、3，转成整数后就是分组标识。

内容的提问来源于stack exchange，提问作者Mark K.