pandas.dataframe中如何根据条件获取元素所在的位置

小编给大家分享一下pandas.dataframe中如何根据条件获取元素所在的位置,相信大部分人都还不怎么了解,因此分享这篇文章给大家参考一下,希望大家阅读完这篇文章后大有收获,下面让我们一起去了解一下吧!

在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中如何根据条件获取元素所在的位置”这篇文章的所有内容,感谢各位的阅读!相信大家都有了一定的了解,希望分享的内容对大家有所帮助,如果还想学习更多知识,欢迎关注行业资讯频道!