падение ряда панды со списком аргументов

Question

Я хотел бросить все строки, которые не находятся в списке в панд DataFrame

Например, рассмотрим этот dataframe:

data = {'name': ['Jason', 'Molly', 'Tina', 'Jake', 'Amy'], 
    'year': [2012, 2012, 2013, 2014, 2014], 
    'reports': [4, 24, 31, 2, 3]} 
df = pd.DataFrame(data, index = ['Cochice', 'Pima', 'Santa   Cruz', 'Maricopa', 'Yuma']) 
df 

уронить строку по названию легко:

df = df[df.name != 'Tina'] # to drop the row which include Tina in the name column 

Но если я хочу сохранить только строку Джейсон и Молли:

List=['Jason', 'Molly'] 
df = df[df.name not in List] 

не работает!

Answer 1

Использование isin и передать список как арг и инвертировать условие с использованием ~:

In [58]: 
names = ['Jason', 'Molly'] 
df[~df['name'].isin(names)] 

Out[58]: 
        name reports year 
Santa   Cruz Tina  31 2013 
Maricopa   Jake  2 2014 
Yuma     Amy  3 2014 

Answer 2

Использование isin с инвертированием булево маску ~. Я думаю, что лучше использовать другое название списка как List, например. l или names, как использование EdChum:

List=['Jason', 'Molly'] 
df = df[~df.name.isin(List)] 

l=['Jason', 'Molly'] 
df = df[~df.name.isin(l)] 

print (df) 
        name reports year 
Santa   Cruz Tina  31 2013 
Maricopa   Jake  2 2014 
Yuma     Amy  3 2014 

Объяснение:

Вы можете использовать mask с инвертированием boolean с ~ с boolean indexing:

print (df['name'].isin(l)) 
Cochice    True 
Pima     True 
Santa   Cruz False 
Maricopa    False 
Yuma     False 
Name: name, dtype: bool 

print (~df['name'].isin(l)) 
Cochice    False 
Pima     False 
Santa   Cruz  True 
Maricopa    True 
Yuma     True 
Name: name, dtype: bool 

Answer 3

isin велик. Другой способ использования query будет

In [821]: List = ['Jason', 'Molly'] 

In [822]: df.query('name not in @List') 
Out[822]: 
        name reports year 
Santa   Cruz Tina  31 2013 
Maricopa   Jake  2 2014 
Yuma     Amy  3 2014