#посчитаем и добавим в таблицу этаж квартиры; варианты — первый, последний, другой
#напишем функцию чтобы распределить значения по категориям
floors_total = real_estate_data['floors_total']
def categorizer_floors(floor):
    if floor == floors_total:
        return 'last'
    if floor == 1:
        return 'first'
    return 'other'

real_estate_data['floor_category'] = real_estate_data['floor'].apply(categorizer_floors)