欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

Python Pandas中Dataframe对象,如何根据列值筛选满足条件的行,并且获得行的索引值

程序员文章站 2022-07-12 14:11:56
...

Python Pandas中Dataframe对象,如何根据列值筛选满足条件的行,并且返回列的索引值,我们举栗子说明一下:

1.先创建一个dataframe变量df:

df = pd.DataFrame(np.arange(16).reshape(4,4), 
                 #index = pd.date_range('20200301', periods=4),
                 index = list('hjkl'),
                 columns = list('ABCD'))
df =df.append(df)

df的样子如下:

Python Pandas中Dataframe对象,如何根据列值筛选满足条件的行,并且获得行的索引值

2.删选满足列条件的行。为获得A列的值等于4, B列的值等于5的所有行的记录,使用如下语句进行筛选

df [ (df['A'] == 4) & (df['B'] == 5)]

结果如下:

Python Pandas中Dataframe对象,如何根据列值筛选满足条件的行,并且获得行的索引值

3.获得满足条件行的索引值

df [ (df['A'] == 4) & (df['B'] == 5)].index.tolist()

输出如下:

Python Pandas中Dataframe对象,如何根据列值筛选满足条件的行,并且获得行的索引值

4.解释。满足条件返回的是一个dataframe对象,可以使用df.index.tolist获取索引值

 

相关标签: python基础