在dataframe中根据一定的条件，得到符合要求的某行元素所在的位置。
代码如下所示：
|
1
2
3
4
5
|
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的不同而不同，这点易于数组中默认的下标。
|
1
2
3
4
5
6
7
|
BoolCol attr 10 1 2220 2 3330 3 2240 3 4450 4 66[30] |
以上这篇pandas.dataframe中根据条件获取元素所在的位置方法(索引)就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持服务器之家。
原文链接：https://blog.csdn.net/xwd18280820053/article/details/72614734
