python - 在熊猫中每 5 行选择前 2 行

Question

我想在熊猫数据框中每 5 行选择前两行。我怎么能这样做？

考虑以下数据框：

col1  | col2 | col3 
1     | 1    | 1
2.    | 2.   | 2
3.    | 3.   | 3 
4     | 4    | 4 
5.    | 5.   | 5 
6.    | 6.   | 6
7.    | 7.   | 7

我想获得以下值

    col1  | col2 | col3 
    1     | 1    | 1
    2.    | 2.   | 2

和

    col1  | col2 | col3
    6.    | 6.   | 6
    7.    | 7.   | 7

我将非常感谢您的帮助！！！

标签： pythonpandas

Answer 1

在索引上使用地板除法，并groupby在其上使用head：

df.groupby(df.index//5).head(2)