Функция к элементу, Python

Представим есть список:

Transports = ("Самолет ТУ", "Автомобиль БМВ" , "Автомобиль Феррари", 
              "Самолет МИГ", "Дирижабль КЛ2", "Автомобиль Форд", "Самокат")

Нужно создать еще один список в том же порядке, например, с кол-вом колес каждого элемента.

Вот мой код:

def wheels(x):
    if re.search('Автомобиль', transport):
        wheelin += "4 колеса"
    elif re.search('Самолет', transport):
        wheelin += "Шасси - 2 колеса"
    else:
        wheelin += "   "
    return wheelin

print (list(map(wheels, transports))

Но на выходе вместо такого:

(Шасси - 2 колеса, 4 колеса, 4 колеса, Шасси - 2 колеса,"   ", 4 колеса,"   ")

Получаем такое:

("Самолет ТУ"Шасси - 2 колеса, "Автомобиль БМВ" Шасси - Шасси - 2 колеса , 
 "Автомобиль Феррари" Шасси - 2 колеса, "Самолет МИГ"Шасси - 2 колеса, 
 "Дирижабль КЛ2"Шасси - 2 колеса, "Автомобиль Форд"Шасси - 2 колеса, 
 "Самокат"Шасси - 2 колеса)

Что не так? Думаю где то в функции что-то не так намудрил, менял и там под разные значения...


Ответы (0 шт):