如何将pandas布尔Series中值为True的对应索引提取为列表

实现方法 #

你当前得到的omgangar是值为布尔类型的Pandas Series，索引为对应的omg取值，直接通过布尔过滤取索引转列表即可：

# 筛选值为True的条目后取索引转列表
result = omgangar[omgangar].index.tolist()

执行后result就是你需要的[2937, 2984, 3231]。

如果要合并进原有代码逻辑，完整写法如下：

omgangar = (matcher[
(matcher['rank'] >= 1) & 
(matcher['rank'] <= 1) & 
(matcher['odds'] >= oddsrank1) & 
(matcher['odds'] <= oddsrank1)]
['omg'].value_counts() == 1)
# 生成目标列表
target_omg_list = omgangar[omgangar].index.tolist()

内容的提问来源于stack exchange，提问作者OldSport