Premium problem56. Filter Rows by Membership

Easy Locked

Return only the rows of df where animal is either "cat" or "dog".

Input

  animal  age  visits priority
a    cat  2.5       1      yes
b    cat  3.0       3      yes
c  snake  0.5       2       no
d    dog  NaN       3      yes
e    dog  5.0       2       no
f    cat  2.0       3       no
g  snake  4.5       1       no
h    cat  NaN       1      yes
i    dog  7.0       2       no
j    dog  3.0       1       no

Output

  animal  age  visits priority
a    cat  2.5       1      yes
b    cat  3.0       3      yes
d    dog  NaN       3      yes
e    dog  5.0       2       no
f    cat  2.0       3       no
h    cat  NaN       1      yes
i    dog  7.0       2       no
j    dog  3.0       1       no

Premium problem

This one's part of Premium. Unlock the full Pandas track plus every other premium problem on the site.

Implement solve(...)