python - 使用索引和值过滤 Pandas 系列

Question

是否有一种干净的方法来使用自定义函数来过滤 Pandas Series，该函数将索引和值作为输入？

这是一段代码，可以实现我想要做的事情:

series = pd.Series({"id5":88, "id3":40})
def custom(k,v):
    if k=="id5":
        return v>20
    else:
        return v>50

filtered_indexes = []
filtered_values = []
for k,v in series.iteritems():
    if custom(k,v):
        filtered_indexes.append(k)
        filtered_values.append(v)
filtered_series = pd.Series(data=filtered_values, index=filtered_indexes)

我的问题是:使用类似的语法可以更清晰和/或更有效地实现同样的目标

series.filter(lambda x: custom(x.index, x.value))

Answer 1

有问题Series.apply have no accesses索引和DataFrame.filter未针对系列实现。

这是可能的，但需要创建DataFrame:

s = series[series.to_frame().apply(lambda x: custom(x.name, x), axis=1).squeeze()]
print (s)
id5    88
dtype: int64

或者使用 groupby 和 filtration :

s = series.groupby(level=0).filter(lambda x: custom(x.name, x)[0])
print (s)
id5    88
dtype: int64