julia - 我无法直接使用 `subset` 在 Julia 的 DataFrames 中使用 `in` 过滤许多值

Question

想知道为什么在使用“子集”时不能使用“in”。 “过滤器”可以使用它。

using DataFrames

df=DataFrame( a=["cat","dog","tiger","parrot","cat","parrot"],
             r=[2,5,1,3,6,3])

subset(df, :a => a->a in["cat", "parrot"], :r => r->r.>1) ##get an error

#but

subset(df, :a => ByRow(a->a in ["cat", "parrot"]), :r => r->r.>1) 

#or

filter([:a,:r]=>(a,r) -> a in ["cat", "parrot"] &&  r.>1, df)

#Thanks.

Answer 1

文档内容如下:

help?> subset
... All transformations must produce vectors containing true or false.

因此你可以这样做

julia> subset(df, :a => a -> in.(a, Ref(["cat", "parrot"])), :r => r -> r .> 1 )

或

julia> subset(df, [:a, :r] => (a,r) -> (in.(a, Ref(["cat", "parrot"]))) .& (r .> 1) )

请注意，如果列表很长，您可能希望将 ["cat", "parrot"] 包装到一个集合中。