I am trying to filter a dataframe using a list with alphanumerical values. Here is a sample dataframe below:

I created a list named "codes" to:

codes = ['D0102', 'D5175']

The code I used is shown below and the dataframe didn't pull up anything:

new_df= df.loc[df['A'].isin(codes)]

The desired output dataframe that I want to display is this:

When I changed the list named "codes" to:

codes = ['Patients', 'DaysWorked']

I used the same line of code:

new_df= df.loc[df['A'].isin(codes)]

The new dataframe pulled the correct data:

CodePudding user response：

There's a good chance that the column A of your df might have some unwanted whitespaces in it. You might wanna do this:

df.loc[df['A'].str.strip().isin(codes)]