1. 程式人生 > >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]