pandas根据列的指定内容选择出行

数据集:

 

train=pd.read_csv('./1.csv')//利用相对路径读取文件内容。

print(train['id']==1)//满足条件的行为True，不满足条件的行为False。

输出为:

0 False
1 False
2 True
3 True

print(train.loc[train['id']==1])//loc根据一组布尔数组来获取行或列。

 输出为:

id sd
2 1 1
3 1 1

相当于选出满足列条件的行组成一个新dataframe。