下面是一个使用Pandas的代码示例,按照另一列分组并提取一列的前几个值:
import pandas as pd
# 创建一个示例数据集
data = {'group': ['A', 'A', 'B', 'B', 'B', 'C'],
'value': [1, 2, 3, 4, 5, 6]}
df = pd.DataFrame(data)
# 按照group列进行分组,并提取value列的前两个值
grouped = df.groupby('group')['value'].apply(lambda x: x.head(2)).reset_index()
print(grouped)
输出结果:
group level_1 value
0 A 0 1
1 A 1 2
2 B 2 3
3 B 3 4
4 B 4 5
5 C 5 6
在上面的代码中,我们首先创建了一个示例数据集df,包含两列:group和value。然后,我们使用groupby
函数按照group列进行分组,并使用apply
函数提取每个分组的value列的前两个值。最后,我们使用reset_index
函数重置索引,并将结果存储在新的DataFrame中。