Используя генерацию списков‚ можно компактно переписать этот код‚ превратив его в одну строку:
squared = [n * n for n in numbers]
Кроме того, можно добавить внутри условие if, что позволит обрабатывать только те элементы, которые соответствуют условию:
squared = [n * n for n in numbers if n > 3]
Машинное обучение. Портфолио реальных проектов
·
Алексей Григорьев