Загрузка данных


import pandas as pd
from animals import init_animals

def to_codes(animals):
    # Преобразуем список животных в pandas.Series
    animal_series = pd.Series(animals)
    
    # Используем метод factorize для кодирования категорий
    codes, uniques = pd.factorize(animal_series)
    
    # Создаем словарь из уникальных названий и их кодов
    animal_dict = dict(zip(uniques, range(len(uniques))))
    
    return animal_dict

animals = init_animals()
print(to_codes(animals))