python - `Pandas argmax` 在屏蔽后获取所有 `True` 索引(Python 3)(例如 (pd.Series > 0).argmax()))

Question

我想要做什么，但 argmax 只给我第一个值为 True 的值:

Se = pd.Series(np.arange(6), index=list("abcdef"))
#a    0
#b    1
#c    2
#d    3
#e    4
#f    5
#dtype: int64

mask = (Se % 2 == 0)
#a     True
#b    False
#c     True
#d    False
#e     True
#f    False
#dtype: bool

mask.argmax()
#'a'

我必须做什么:

Se[mask].index
# Index(['a', 'c', 'e'], dtype='object')

这并不算太不方便，但我必须首先实例化Series，这降低了我的工作效率。如果能够做到这一点那就太好了:

(pd.Series(np.arange(6), index=list("abcdef")) % 2 == 0).argmax()

我的问题是:如何使用 argmax 执行此操作？如果这无法通过 argmax 完成，我可以使用 pandas 中的不同函数来完成此操作吗？

Answer 1

您可以使用compress :

idx = pd.Series(np.arange(6), index=list("abcdef")).compress(lambda x: x % 2 == 0).index

结果输出:

Index(['a', 'c', 'e'], dtype='object')