我有一個資料集：
id value
a1 14
a1 2
a1 34
a1 11
a1 78
b1 11
b1 9
b1 6
我想按每個組的行數過濾該資料集，使其不高于 4。因此所需的輸出將是：
id value
a1 14
a1 2
a1 34
a1 11
b1 11
b1 9
b1 6
怎么做？
uj5u.com熱心網友回復：
您可以使用groupby.head：
out = df.groupby('id').head(4)
如果你有 pandas >=1.4.0，那么你也可以使用groupby.nth切片：
out = df.groupby('id').nth[:4]
輸出
id value
0 a1 14
1 a1 2
2 a1 34
3 a1 11
5 b1 11
6 b1 9
7 b1 6
轉載請註明出處，本文鏈接：https://www.uj5u.com/ruanti/466430.html
標籤：Python python-3.x 熊猫 数据框 筛选