Să reunim totul (2)
Uneori facem greșeli când apelăm funcții – chiar și pe cele pe care le-am scris noi înșine. Nu-i nimic! În acest exercițiu, vei îmbunătăți funcția count_entries() din capitolul anterior, adăugând un bloc try-except. Astfel, funcția va afișa un mesaj util atunci când utilizatorul furnizează un nume de coloană care nu există în DataFrame.
Pentru comoditate, pandas a fost importat ca pd, iar fișierul 'tweets.csv' a fost încărcat în DataFrame-ul tweets_df. O parte din codul tău anterior este deja furnizat.
Acest exercițiu face parte din cursul
Introducere în funcții în Python
Instrucțiuni pentru exercițiu
- Adaugă un bloc
tryastfel încât, atunci când funcția este apelată cu argumentele corecte, să proceseze DataFrame-ul și să returneze un dicționar cu rezultatele. - Adaugă un bloc
exceptastfel încât, atunci când funcția este apelată incorect, să afișeze următorul mesaj de eroare:'The DataFrame does not have a ' + col_name + ' column.'.
Exercițiu interactiv practic
Încearcă acest exercițiu completând acest cod de exemplu.
# 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)