Включения Python

Включения Python — это уникальный способ обработки данных, который позволяет создавать новую структуру последовательности данных из одной последовательности данных.

Включения Python — это мощный и лаконичный синтаксис, подходящий для создания списков, словарей, множеств и генераторов.

При использовании включений нужно обращать внимание на читаемость и стараться сохранять выражения краткими, чтобы не влиять на читаемость и поддерживаемость кода.

Python поддерживает включения для различных структур данных:

  • Списковое включение
  • Словарное (dict) включение
  • Включение множества
  • Включение кортежа

Списковое включение

Формат спискового включения:

[表达式 for 变量 in 列表] 
[out_exp_res for out_exp in input_list]

或者 

[表达式 for 变量 in 列表 if 条件]
[out_exp_res for out_exp in input_list if condition]
  • out_exp_res: выражение для генерации элементов списка, которое может быть функцией с возвращаемым значением.
  • for out_exp in input_list: перебирает input_list и передаёт out_exp в выражение out_exp_res.
  • if condition: условный оператор, который может отфильтровать значения в списке, не удовлетворяющие условию.

Отфильтровать списки строк, длина которых меньше или равна 3, и преобразовать оставшиеся в заглавные буквы:

Пример

>>> names = ['Bob','Tom','alice','Jerry','Wendy','Smith']
>>> new_names = [name.upper()for name in names if len(name)>3]
>>> print(new_names)
['ALICE', 'JERRY', 'WENDY', 'SMITH']

Вычислите целые числа в пределах 30, которые делятся на 3:

Пример

>>> multiples = [i for i in range(30) if i % 3 == 0]
>>> print(multiples)
[0, 3, 6, 9, 12, 15, 18, 21, 24, 27]

Словарное включение

Базовый формат словарного включения:

{ key_expr: value_expr for value in collection }

или 

{ key_expr: value_expr for value in collection if condition }

Создайте словарь, используя строки и их длины:

Пример

listdemo = ['Google','Example', 'Taobao']
# Используйте строки из списка как ключи, а длины строк — как значения, чтобы сформировать пары ключ-значение
>>> newdict = {key:len(key) for key in listdemo}
>>> newdict
{'Google': 6, 'Example': 6, 'Taobao': 6}

Укажите три числа, используйте эти три числа как ключи, а квадраты этих трёх чисел как значения, чтобы создать словарь:

Пример

>>> dic = {x: x**2 for x in (2, 4, 6)}
>>> dic
{2: 4, 4: 16, 6: 36}
>>> type(dic)
<class 'dict'>

Генератор множеств

Базовый формат генератора множеств:

{ expression for item in Sequence }
или 
{ expression for item in Sequence if conditional }

Вычислите квадраты чисел 1, 2, 3:

Пример

>>> setnew = {i**2 for i in (1,2,3)}
>>> setnew
{1, 4, 9}

Определите буквы, которые не входят в "abc", и выведите их:

Пример

>>> a = {x for x in 'abracadabra' if x not in 'abc'}
>>> a
{'d', 'r'}
>>> type(a)
<class 'set'>

Кортежное включение (генераторное выражение)

Кортежное включение может использовать такие типы данных, как диапазоны, кортежи, списки, словари и множества, чтобы быстро сгенерировать кортеж, удовлетворяющий заданным требованиям.

Основной формат кортежного включения:

(expression for item in Sequence )
или 
(expression for item in Sequence if conditional )

Использование кортежного включения точно такое же, как и списковое включение, за исключением того, что кортежное включение использует()круглые скобки для обрамления каждой части, тогда как списковое включение использует квадратные скобки[], кроме того, результат, возвращаемый кортежным включением, — это объект-генератор.

Например, мы можем использовать следующий код, чтобы сгенерировать кортеж, содержащий числа 1-9:

Пример

>>> a = (x for x in range(1,10))
>>> a
<generator object <genexpr> at 0x7faf6ee20a50>  # Возвращаемое значение — это объект-генератор

>>> tuple(a)       # Используя функцию tuple(), можно напрямую преобразовать объект-генератор в кортеж
(1, 2, 3, 4, 5, 6, 7, 8, 9)
Другие расширения