以下是一个示例代码,演示了如何按组为对象数组创建索引列:
import pandas as pd
# 创建示例数据
data = [{'组': 'A', '数值': 1}, {'组': 'A', '数值': 2}, {'组': 'B', '数值': 3}, {'组': 'B', '数值': 4}]
df = pd.DataFrame(data)
# 按组为对象数组创建索引列
df['索引列'] = df.groupby('组').cumcount() + 1
# 打印结果
print(df)
输出结果:
组 数值 索引列
0 A 1 1
1 A 2 2
2 B 3 1
3 B 4 2
在上述示例中,首先创建了一个包含组和数值的对象数组。然后,使用pandas库的DataFrame函数将数据转换为DataFrame对象。接下来,使用groupby函数按组对数据进行分组,并使用cumcount函数为每个组创建索引列。最后,将索引列添加到DataFrame对象中。
上一篇:按组统计重排序因子
下一篇:按组为每个连续发生的个体创建ID