Чтобы сгладить список в Python с помощью itertools, вы можете использовать функцию itertools.chain.from_iterable(). Вот пример:
import itertools
def flatten_list(lst):
flattened = list(itertools.chain.from_iterable(lst))
return flattened
# Example usage
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = flatten_list(nested_list)
print(flattened_list)
Выход:
[1, 2, 3, 4, 5, 6, 7, 8, 9]
Этот код использует функцию itertools.chain.from_iterable()для выравнивания lstпутем объединения всех подсписков. Полученный плоский список затем преобразуется обратно в обычный список с помощью функции list().
Альтернативный метод.
Другой способ сгладить список в Python — использовать понимание списка с вложенными циклами. Вот пример:
def flatten_list(lst):
flattened = [item for sublist in lst for item in sublist]
return flattened
# Example usage
nested_list = [[1, 2, 3], [4, 5, 6], [7, 8, 9]]
flattened_list = flatten_list(nested_list)
print(flattened_list)
Выход:
[1, 2, 3, 4, 5, 6, 7, 8, 9]
В этом коде генератор списка перебирает каждый подсписок в lst, а затем перебирает каждый элемент в подсписке, выравнивая список в процессе.