是否可以在pandas中使用自定义过滤器功能？

Question

我可以用我的助手函数来确定一个镜头是否是三个指针作为Pandas的过滤函数？我的实际函数要复杂得多，但是为了这个问题我简化了它。在

def isThree(x, y):
    return (x + y == 3)

print data[isThree(data['x'], data['y'])].head()

Answer 1

是的，只要您的函数返回一个具有相同索引的布尔序列，您就可以用输出对原始数据帧进行切片。在这个简单的示例中，我们可以将Series传递给函数：

import pandas as pd
import numpy as np

df = pd.DataFrame(np.random.randint(0, 4, (30, 2)))
def isThree(x, y):
    return x + y == 3

df[isThree(df[0], df[1])]
#    0  1
#2   2  1
#5   2  1
#9   0  3
#11  2  1
#12  0  3
#13  2  1
#27  3  0