В Python вы часто можете столкнуться с ситуациями, когда вам нужно сгладить список списков, преобразовав его в одномерный список. Эта операция довольно распространена и может быть выполнена с использованием различных методов. В этой статье мы рассмотрим возможности модуля Python itertools и изучим несколько методов выравнивания списка списков. Итак, давайте углубимся и разгадаем магию itertools!

Метод 1: использование генератора списков
Один простой способ сгладить список списков — использовать генератор списков. Вот пример:

nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = [item for sublist in nested_list for item in sublist]
print(flattened_list)

Выход:

[1, 2, 3, 4, 5, 6, 7, 8, 9]

Метод 2: использование itertools.chain()
Модуль itertools Python предоставляет мощные инструменты для эффективной итерации. Функцию chain()из itertools можно использовать для выравнивания списка списков:

import itertools
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = list(itertools.chain(*nested_list))
print(flattened_list)

Выход:

[1, 2, 3, 4, 5, 6, 7, 8, 9]

Метод 3: использование itertools.chain.from_iterable()
Еще одна удобная функция itertools — chain.from_iterable(). Этот метод упрощает процесс выравнивания:

import itertools
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = list(itertools.chain.from_iterable(nested_list))
print(flattened_list)

Выход:

[1, 2, 3, 4, 5, 6, 7, 8, 9]

Метод 4. Используя оператор и sum()
Используя оператори функцию sum(), мы можем добиться выравнивания списка:

nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = sum(nested_list, [])
print(flattened_list)

Выход:

[1, 2, 3, 4, 5, 6, 7, 8, 9]

Метод 5: использование itertools.flatten() (Python 3.10+)
Начиная с Python 3.10, модуль itertools представляет экспериментальную функцию flatten(), специально разработанную для этой цели:

import itertools
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = list(itertools.flatten(nested_list))
print(flattened_list)

Выход:

[1, 2, 3, 4, 5, 6, 7, 8, 9]

Сглаживание списков — обычная операция в Python, и модуль itertools предоставляет несколько эффективных методов для выполнения этой задачи. В этой статье мы рассмотрели различные методы, включая понимание списков, itertools.chain(), itertools.chain.from_iterable(), оператор * с sum() и экспериментальную функцию itertools.flatten(), представленную в Python 3.10. Теперь у вас есть широкий выбор вариантов, исходя из ваших конкретных потребностей. Приятного кодирования!