python布尔系列_python – 获取pandas布尔系列为True的索引列表

使用

Boolean Indexing

>>> s = pd.Series([True, False, True, True, False, False, False, True])

>>> s[s].index

Int64Index([0, 2, 3, 7], dtype='int64')

如果需要np.array对象,请获取.values

>>> s[s].index.values

array([0, 2, 3, 7])

>>> np.nonzero(s)

(array([0, 2, 3, 7]),)

>>> np.flatnonzero(s)

array([0, 2, 3, 7])

>>> np.where(s)[0]

array([0, 2, 3, 7])

>>> np.argwhere(s).ravel()

array([0, 2, 3, 7])

>>> s.index[s]

array([0, 2, 3, 7])

使用python的内置filter

>>> [*filter(s.get, s.index)]

[0, 2, 3, 7]

使用列表理解

>>> [i for i in s.index if s[I]]

[0, 2, 3, 7]