pandas groupby 分组取每组的前几行记录

pandas groupby分组取每组的前几行记录

dic = {'地市': ['廊坊','廊坊','廊坊','张家口','张家口','张家口','廊坊','承德','承德','承德','石家庄','石家庄','石家庄','石家庄'],'组号':[1,2,1,2,1,2,1,2,1,2,1,2,1,2],'数量':[11,12,13,14,15,16,17,18,19,20,21,22,23,24]}
p_city = pd.DataFrame(dic)
print(p_city )

输出：

grouped = p_city.groupby(['地市']).head(2)
print(grouped)

输出：