python - 每 7 行中，获取第 n 行 pandas

Question

我有一个像这样的 df，大约有 1000 行:

      0        1
0  1.345    2.456
1  2.123    3.564
2  0.023    3.548
3  3.457    2.456
4  1.754    3.564
5  0.905    3.548
6  3.674    7.543
7  9.443    6.4433...

它的组织方式是每 7 行构成一组数据(此处数据无法排序)。在 7 行的每个“组”中，我想获得第一行，以便我的新数据框看起来像:

      0        1
0  1.345    2.456
7  9.443    6.4433

我可以通过创建一个重复 1-7 的新列并仅按该列进行过滤来解决它...

      0        1    groupby_col
0  1.345    2.456        1
1  2.123    3.564        2
2  0.023    3.548        3
3  3.457    2.456        4
4  1.754    3.564        5
5  0.905    3.548        6
6  3.674    7.543        7
7  9.443    6.4433       1...

然后...

df[df['groupby_col'] == 1]

有没有一种方法可以在 pandas 中执行此操作而无需创建额外的列然后进行过滤？

Answer 1

选项 1:

In [54]: df.iloc[::7]
Out[54]:
       0       1
0  1.345  2.4560
7  9.443  6.4433

选项 2:

In [53]: df.iloc[np.arange(len(df))%7==0]
Out[53]:
       0       1
0  1.345  2.4560
7  9.443  6.4433