python - 获取有条件的索引

Question

如何找到 search_value 的所有条目的位置(索引)？

import pandas as pd
import numpy as np
search_value=8
lst=[5, 8, 2, 7, 8, 8, 2, 4]
df=pd.DataFrame(lst)
df["is_search_value"]=np.where(df[0]==search_value, True, False)
print(df.head(20))

输出:

   0  is_search_value
0  5            False
1  8             True
2  2            False
3  7            False
4  8             True
5  8             True
6  2            False
7  4            False

理想的输出:

1 4 5

如果 search_value 是 10 而不是理想的输出:

None

Answer 1

您可以在条件列表理解中使用枚举来获取索引位置。

my_list = [5, 8, 2, 7, 8, 8, 2, 4]
search_value = 8
>>> [i for i, n in enumerate(my_list) if n == search_value]
[1, 4, 5]

如果搜索值不在列表中，则返回一个空列表(不完全是 None，但仍然是一个错误)。

使用 pandas，您可以使用 bool 索引来获取匹配项，然后将索引提取到列表中:

df[df[0] == search_value].index.tolist()

使用空列表将满足 None 的条件(它们的计算结果均为 False)。如果你真的不需要，那么使用@cᴏʟᴅsᴘᴇᴇᴅ的建议。