Skip to content
Advertisement

Check if value of dataframe is existing in list

I am trying to split a pandas Dataframe into two, based on the value in the column “country”.

If the value exists in the following list (EU-COUNTRY-CODES), the row should be added to a dataframe called “EU”, if it does not exist in the list I want to add the row to another dataframe.

eu-country-codes = ["BE","BG","DK","EE","FI","FR","GR","IE","IT","HR","LV","LT","LU","MT","NL","AT","PL","PT","RO","SM","SE","SK","SI","ES","CZ","HU","VA","FO","CY"]

I tried to do it this way but got following Error:

data = pd.read_csv("datei.csv", sep=",", header=0)
eu= data[data["country"] not in eu-country-codes]

ValueError: The truth value of a Series is ambiguous. Use a.empty, a.bool(), a.item(), a.any() or a.all().

Advertisement

Answer

try this:

data[data["country"].isin( eu-country-codes)==False]
User contributions licensed under: CC BY-SA
5 People found this is helpful
Advertisement