Python: Suppression de lignes de dataframe dont la valeur n'existe pas dans le dictionnaire

Question

j'ai un fichier csv (ou dataframe) comme ci-dessous:

Text Location State 
A Florida, USA Florida 
B NY    New York 
C  
D abc 

Et un dictionnaire avec une paire de valeurs clés comme:

stat_map = { 
     'FL': 'Florida', 
     'NY': 'New York', 
     'AR': 'Arkansas', 
} 

Comment puis-je supprimer les rangées 3 et 4, c.-à-d. La ligne avec le texte C & D afin que mon fichier ne contienne que les lignes pour lesquelles j'ai une valeur dans le dictionnaire. Toutes les lignes pour lesquelles l'état est vide ou dont la valeur n'est pas dans la valeur du dictionnaire doivent être supprimées. La sortie finale devrait ressembler à:

Text Location State 
    A Florida, USA Florida 
    B NY    New York 

S'il vous plaît aider.

Answer 1

Utilisez extract + replace, dernière supprimer des lignes par dropna:

stat_map = { 
     'FL': 'Florida', 
     'NY': 'New York', 
     'AR': 'Arkansas', 
} 

#get list from all values from keys and values of dict 
L = list(stat_map.keys()) + list(stat_map.values()) 
print (L) 
['NY', 'FL', 'AR', 'New York', 'Florida', 'Arkansas'] 


df['State1'] = df['Location'].str.extract('(' + '|'.join(L) + ')', expand=False) 
          .replace(stat_map) 
df = df.dropna(subset=['State1']) 
print (df) 
    Text  Location  State State1 
0 A Florida, USA Florida Florida 
1 B   NY New York New York