Объединяем всё вместе (2)

Иногда мы допускаем ошибки при вызове функций — даже тех, которые написали сами. Но не переживайте! В этом упражнении вы улучшите функцию count_entries() из предыдущей главы, добавив в неё блок try-except. Это позволит функции выводить понятное сообщение об ошибке, если пользователь передаст имя столбца, которого нет в DataFrame.

Как и прежде, для вашего удобства библиотека pandas импортирована как pd, а файл 'tweets.csv' загружен в DataFrame tweets_df. Часть кода из предыдущей работы также уже предоставлена.

Это упражнение является частью курса

Введение в функции Python

Посмотреть курс

Инструкции к упражнению

Добавьте блок try, чтобы при вызове функции с корректными аргументами она обрабатывала DataFrame и возвращала словарь с результатами.
Добавьте блок except, чтобы при некорректном вызове функция выводила следующее сообщение об ошибке: 'The DataFrame does not have a ' + col_name + ' column.'.

Интерактивное практическое упражнение

Попробуйте выполнить это упражнение, дополнив этот пример кода.

# 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)

Редактировать и запускать код