Mise en pratique : tout rassembler (2)

Il arrive qu'on se trompe en appelant des fonctions — même celles que vous avez écrites. Pas de souci ! Dans cet exercice, vous allez améliorer votre travail précédent avec la fonction count_entries() du dernier chapitre en y ajoutant un bloc try-except. Ainsi, votre fonction pourra afficher un message utile lorsque l'on appelle count_entries() avec un nom de colonne qui n'existe pas dans le DataFrame.

Encore une fois, pour vous simplifier la vie, pandas a été importé sous le nom pd et le fichier 'tweets.csv' a été importé dans le DataFrame tweets_df. Des parties du code provenant de votre travail précédent sont aussi fournies.

Cette activité fait partie du cours

Introduction aux fonctions en Python

Voir le cours

Instructions de l’exercice

Ajoutez un bloc try afin que, lorsque la fonction est appelée avec les bons arguments, elle traite le DataFrame et retourne un dictionnaire de résultats.
Ajoutez un bloc except afin que, lorsque la fonction est appelée de façon incorrecte, elle affiche le message d'erreur suivant : 'The DataFrame does not have a ' + col_name + ' column.'.

Exercice interactif pratique

Essayez cet exercice en complétant ce code d’exemple.

# Define count_entries()
def count_entries(df, col_name='lang'):
    """Return a dictionary with counts of
    occurrences as value for each key."""

    # Initialize an empty dictionary: cols_count
    cols_count = {}

    # Add try block
    ____:
        # Extract column from DataFrame: col
        col = df[col_name]
        
        # Iterate over the column in DataFrame
        for entry in col:
    
            # If entry is in cols_count, add 1
            if entry in cols_count.keys():
                cols_count[entry] += 1
            # Else add the entry to cols_count, set the value to 1
            else:
                cols_count[entry] = 1
    
        # Return the cols_count dictionary
        return cols_count

    # Add except block
    ____:
        ____

# Call count_entries(): result1
result1 = count_entries(tweets_df, 'lang')

# Print result1
print(result1)

Modifier et exécuter le code