python数据框切片_python – 如何根据列值切片pandas数据框？

在您的示例数据集上,以下工作：

In [35]:

df.iloc[0:df[df.year == 'y3'].index[0]]

Out[35]:

year col1

0 y1 val_1

1 y1 val_2

2 y1 val_3

3 y2 val_4

4 y2 val_5

5 y2 val_6

所以打破这个,我们执行一个布尔索引来查找等于年值的行：

In [36]:

df[df.year == 'y3']

Out[36]:

year col1

6 y3 val_7

7 y3 val_8

8 y3 val_9

但我们对索引很感兴趣,所以我们可以用它来切片：

In [37]:

df[df.year == 'y3'].index

Out[37]:

Int64Index([6, 7, 8], dtype='int64')

但是我们只需要切片的第一个值因此调用index [0],但是如果你已经按年值排序,那么只需执行df [df.year< y3]会更简单,更有效.