在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中根據條件獲取元素所在的位置方法(索引)就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答