python - 如何获取过滤后的 Panda 系列的索引？

Question

我正在使用 13548 个连续值，我的数据是按以下方式构建的:

s = pd.Series(delta, index=st)
s[:10]

GEOM_190_190 0.00
GEOM_190_192 1.91
GEOM_190_194 2.54
GEOM_190_196 4.90
GEOM_190_198 6.03
GEOM_190_200 6.92
GEOM_190_202 8.60
GEOM_190_204 10.06
GEOM_190_206 11.34
GEOM_190_208 12.43
数据类型:float64

我应用了以下过滤器:
extra = [i for i in s if i<52.55]
extra2 = [i for i in s if i>61 and i<68]
extra3 = [i for i in s if i>73]

我想知道这些值的索引。 有更好的方法吗？

谢谢!

Answer 1

您可以使用loc访问一组行并返回另一个系列，然后您可以使用它来检查索引。

import pandas as pd

test = pd.Series({
    1: 5.0,
    2: 4.5,
    3: 8.2,
    4: 9.0,
    5: 6.7
})

extra = test.loc[lambda x: x < 8]
# or extra = test[test < 8] # more concise but potentially slower

print(extra)
# 1    5.0
# 2    4.5
# 5    6.7
# dtype: float64

有几种方法可以做到这一点，并且性能可能会有所不同，如所述 here 。

编辑:要在一个 loc 中执行多个条件，请尝试如下操作:

test.loc[(test > 5) & (test < 8)]
# 5    6.7
# dtype: float64