Všechno dohromady (2)
Při volání funkcí se občas stane chyba – i u těch, které jsi napsal/a sám/sama. Nic si z toho nedělej! V tomto cvičení vylepšíš svou předchozí práci s funkcí count_entries() z minulé kapitoly tím, že do ní přidáš blok try-except. Díky tomu funkce zobrazí užitečnou zprávu, když ji zavoláš se jménem sloupce, který v DataFrame neexistuje.
Pro pohodlí je pandas opět importován jako pd a soubor 'tweets.csv' je načten do DataFrame tweets_df. Části kódu z předchozího cvičení jsou také k dispozici.
Toto cvičení je součástí kurzu
Úvod do funkcí v Pythonu
Pokyny k cvičení
- Přidej blok
trytak, aby funkce při správném zavolání zpracovala DataFrame a vrátila slovník s výsledky. - Přidej blok
excepttak, aby funkce při nesprávném zavolání zobrazila následující chybovou zprávu:'The DataFrame does not have a ' + col_name + ' column.'.
Interaktivní cvičení na vyzkoušení si v praxi
Vyzkoušejte si toto cvičení dokončením tohoto ukázkového kódu.
# 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)