Python Pandas:如何检索多个值的行索引

时间:2019-07-04 07:20:54

标签: python pandas

我想获取包含某些值(不止一个)的行的行索引。

我尝试将要检索的值放在列表中,但失败了,我只能逐个搜索索引,这很繁琐。

可能但乏味

rows.loc[rows['NO']=='NO'].index.values
rows.loc[rows['NO']=='Sub Total :'].index.values
rows.loc[rows['NO']=='Generated By:'].index.values
rows.loc[rows['NO']=='Product :'].index.values

失败

rows.loc[rows['NO']==list1].index.values

2 个答案:

答案 0 :(得分:1)

尝试

rows.loc[rows['NO'].isin(list1)].index.values

答案 1 :(得分:1)

str.contains用作:

l = ['NO','Sub Total :', 'Generated By:', 'Product :']
index = rows.loc[rows['NO'].str.contains('|'.join(l)), 'NO'].index.values