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

pandas.dataframe中根据条件获取元素所在的位置方法(索引)

程序员文章站 2022-06-11 22:36:55
在dataframe中根据一定的条件,得到符合要求的某行元素所在的位置。 代码如下所示: df = pd.dataframe({'boolcol': [1, 2...

在dataframe中根据一定的条件,得到符合要求的某行元素所在的位置。

代码如下所示:

df = pd.dataframe({'boolcol': [1, 2, 3, 3, 4],'attr': [22, 33, 22, 44, 66]}, 
  index=[10,20,30,40,50]) 
print(df) 
a = df[(df.boolcol==3)&(df.attr==22)].index.tolist() 
print(a) 

df如下所示,以上通过选取“boolcol”取值为3且“attr”取值为22的行,得到该行在df中的位置

注意:返回的位置为index列表,根据index的不同而不同,这点易于数组中默认的下标。

 boolcol attr 
10  1 22 
20  2 33 
30  3 22 
40  3 44 
50  4 66 
[30] 

以上这篇pandas.dataframe中根据条件获取元素所在的位置方法(索引)就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。