Python Filter Dataframe with an unknown list of values with Kwargs

user_13643669 0
Viewed 93

Im new to coding but I wanted to see if there was a way to filter a dataframe with an unknown number of values. What I have so far

df[(df['column'].astype(str)=='one')|(df['column'].astype(str)=='two')|(df['column'].astype(str)=='three')]

This filters a dataframe just fine but I have an unknown number of values. What I wanted would be something like:

x=['one','two','three']
y=['one','two','three','four']
z=['one','two','three','four','five']

for arg in *args:
     df=df[(df['column'].astype(str)==arg )]

But this does not seem to work. Does anyone have any thoughts?

pandas args python
1 Answers

Use isin

>>> df[df['column'].isin(x)]  # or y or z
user_152399510
Related