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)
输出: