Курс Python → Возведение в квадрат с помощью itertools

Для возведения в квадрат всех чисел в заданном диапазоне сначала нам нужно импортировать модуль itertools. Затем мы можем использовать функцию imap из модуля itertools для применения квадратной функции ко всем элементам заданного диапазона. Это позволит нам создать итератор, содержащий квадраты всех чисел в этом диапазоне.


import itertools

# Задаем диапазон чисел
start = 1
end = 5

# Используем функцию imap для нахождения квадратов чисел в диапазоне
squared_numbers = itertools.imap(lambda x: x**2, range(start, end+1))

# Выводим результат
for num in squared_numbers:
    print(num)

В этом примере мы импортируем модуль itertools и задаем начальное и конечное значение диапазона чисел. Затем мы используем функцию imap для применения лямбда-функции, которая возводит число в квадрат, ко всем элементам диапазона. Результатом будет итератор, содержащий квадраты всех чисел в заданном диапазоне.

Наконец, мы перебираем полученный итератор и выводим каждый квадрат на экран. Таким образом, мы можем легко и эффективно получить квадраты всех чисел в заданном диапазоне, используя модуль itertools и функцию imap.

Твои коллеги будут рады, поделись в