Python Dataframe 根据属性值获取所在行的索引
2020-09-21 本文已影响0人
王叽叽的小心情
采用df.index.to_list()进行转换
df = pd.DataFrame({'name':['小王','小李'],'age':[15,16]})
df
Out[164]:
name age
0 小王 15
1 小李 16
index = df.loc[df.name=='小王'].index.to_list()
index
Out[165]: [0]
index = df.loc[(df.age==16) | (df.age==15)].index.to_list()
index
Out[167]: [0, 1]
