Функция к элементу, 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 колеса)
Что не так? Думаю где то в функции что-то не так намудрил, менял и там под разные значения...