Содержание страницы
5. Структуры данных¶Data Structures
В этой главе более подробно рассматриваются некоторые уже изученные темы, а также добавляются новые.
5.1. Подробнее о списках¶More on Lists
Тип данных list имеет ещё несколько методов. Ниже перечислены все методы объектов list:
- list.append(x)
Добавляет элемент в конец списка. Эквивалентно a[len(a):] = [x].
- list.extend(L)
Расширяет список, добавляя все элементы из указанного списка. Эквивалентно a[len(a):] = L.
- list.insert(i, x)
Вставляет элемент в заданную позицию. Первый аргумент – это индекс элемента, перед которым нужно вставить, поэтому a.insert(0, x) вставляет в начало списка, а a.insert(len(a), x) эквивалентно a.append(x).
- list.remove(x)
Удаляет первый элемент из списка, значение которого равно x. Если такой элемент отсутствует, возникает ошибка.
- list.pop([i])
Удаляет элемент в заданной позиции в списке и возвращает его. Если индекс не указан, a.pop() удаляет и возвращает последний элемент списка. (Квадратные скобки вокруг i в сигнатуре метода указывают, что параметр является необязательным, а не на то, что нужно вводить квадратные скобки в этой позиции. Эта нотация часто встречается в Справочнике по библиотеке Python.)
- list.clear()
Удаляет все элементы из списка. Эквивалентно del a[:].
- list.index(x)
Возвращает индекс первого элемента в списке, значение которого равно x. Если такого элемента нет, возникает ошибка.
- list.count(x)
Возвращает количество вхождений x в список.
- list.sort()
Сортирует элементы списка на месте.
- list.reverse()
Переворачивает элементы списка на месте.
- list.copy()
Возвращает поверхностную копию списка. Эквивалентно a[:].
Пример, использующий большинство методов списка:
>>> a = [66.25, 333, 333, 1, 1234.5]
>>> print(a.count(333), a.count(66.25), a.count('x'))
2 1 0
>>> a.insert(2, -1)
>>> a.append(333)
>>> a
[66.25, 333, -1, 333, 1, 1234.5, 333]
>>> a.index(333)
1
>>> a.remove(333)
>>> a
[66.25, -1, 333, 1, 1234.5, 333]
>>> a.reverse()
>>> a
[333, 1234.5, 1, 333, -1, 66.25]
>>> a.sort()
>>> a
[-1, 1, 66.25, 333, 333, 1234.5]
>>> a.pop()
1234.5
>>> a
[-1, 1, 66.25, 333, 333]
Возможно, вы заметили, что такие методы, как insert, remove или sort, которые только изменяют список, не печатают возвращаемое значение – они возвращают значение по умолчанию None. [1] Это принцип проектирования для всех изменяемых структур данных в Python.
5.1.1. Использование списков в качестве стеков¶Using Lists as Stacks
Методы списков позволяют легко использовать список как стек, где последний добавленный элемент извлекается первым (принцип «последним пришёл – первым вышел»). Чтобы добавить элемент на вершину стека, используйте append(). Чтобы извлечь элемент с вершины стека, используйте pop() без явного указания индекса. Например:
>>> stack = [3, 4, 5]
>>> stack.append(6)
>>> stack.append(7)
>>> stack
[3, 4, 5, 6, 7]
>>> stack.pop()
7
>>> stack
[3, 4, 5, 6]
>>> stack.pop()
6
>>> stack.pop()
5
>>> stack
[3, 4]
5.1.2. Использование списков в качестве очередей¶Using Lists as Queues
Список также можно использовать как очередь, где первый добавленный элемент извлекается первым (первым пришёл – первым ушёл); однако списки неэффективны для этой цели. Хотя добавление и извлечение элементов с конца списка выполняется быстро, вставка или извлечение с начала списка – медленно (поскольку все остальные элементы приходится сдвигать на одну позицию).
Для реализации очереди используйте collections.deque, который разработан для быстрого добавления и извлечения элементов с обоих концов. Например:
>>> from collections import deque
>>> queue = deque(["Eric", "John", "Michael"])
>>> queue.append("Terry") # Терри прибывает
>>> queue.append("Graham") # Грэм прибывает
>>> queue.popleft() # Первый прибывший теперь уходит
'Eric'
>>> queue.popleft() # Второй прибывший теперь уходит
'John'
>>> queue # Оставшаяся очередь в порядке прибытия
deque(['Michael', 'Terry', 'Graham'])
5.1.3. Списковые включения¶List Comprehensions
Списковые включения предоставляют краткий способ создания списков. Обычно они используются для создания новых списков, где каждый элемент является результатом некоторых операций, применённых к каждому элементу другой последовательности или итерируемого объекта, или для создания подпоследовательности из элементов, удовлетворяющих определённому условию.
Например, предположим, мы хотим создать список квадратов, как в:
>>> squares = []
>>> for x in range(10):
... squares.append(x**2)
...
>>> squares
[0, 1, 4, 9, 16, 25, 36, 49, 64, 81]
Обратите внимание, что этот способ создаёт (или перезаписывает) переменную с именем x, которая продолжает существовать после завершения цикла. Можно вычислить список квадратов без побочных эффектов, используя:
squares = list(map(lambda x: x**2, range(10)))
или, что то же самое:
squares = [x**2 for x in range(10)]
что более кратко и читаемо.
Списочное выражение (list comprehension) состоит из скобок, содержащих выражение, за которым следует предложение for, а затем ноль или более предложений for или if. Результатом будет новый список, полученный в результате вычисления выражения в контексте предложений for и if, которые следуют за ним. Например, это списочное выражение объединяет элементы двух списков, если они не равны:
>>> [(x, y) for x in [1,2,3] for y in [3,1,4] if x != y]
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]
и это эквивалентно:
>>> combs = []
>>> for x in [1,2,3]:
... for y in [3,1,4]:
... if x != y:
... combs.append((x, y))
...
>>> combs
[(1, 3), (1, 4), (2, 3), (2, 1), (2, 4), (3, 1), (3, 4)]
Обратите внимание, что порядок операторов for и if одинаков в обоих этих фрагментах.
Если выражение является кортежем (например, (x, y) в предыдущем примере), его необходимо заключить в скобки.
>>> vec = [-4, -2, 0, 2, 4]
>>> # создать новый список с удвоенными значениями
>>> [x*2 for x in vec]
[-8, -4, 0, 4, 8]
>>> # отфильтровать список, чтобы исключить отрицательные числа
>>> [x for x in vec if x >= 0]
[0, 2, 4]
>>> # применить функцию ко всем элементам
>>> [abs(x) for x in vec]
[4, 2, 0, 2, 4]
>>> # вызвать метод для каждого элемента
>>> freshfruit = [' banana', ' loganberry ', 'passion fruit ']
>>> [weapon.strip() for weapon in freshfruit]
['banana', 'loganberry', 'passion fruit']
>>> # создать список кортежей из двух элементов вида (число, квадрат)
>>> [(x, x**2) for x in range(6)]
[(0, 0), (1, 1), (2, 4), (3, 9), (4, 16), (5, 25)]
>>> # кортеж должен быть заключён в скобки, иначе возникнет ошибка
>>> [x, x**2 for x in range(6)]
File "<stdin>", line 1, in ?
[x, x**2 for x in range(6)]
^
SyntaxError: invalid syntax
>>> # развернуть список с помощью спискового включения с двумя 'for'
>>> vec = [[1,2,3], [4,5,6], [7,8,9]]
>>> [num for elem in vec for num in elem]
[1, 2, 3, 4, 5, 6, 7, 8, 9]
Списковые включения могут содержать сложные выражения и вложенные функции:
>>> from math import pi
>>> [str(round(pi, i)) for i in range(1, 6)]
['3.1', '3.14', '3.142', '3.1416', '3.14159']
5.1.4. Вложенные списковые включения¶Nested List Comprehensions
Начальное выражение в списковом включении может быть любым произвольным выражением, включая другое списковое включение.
Рассмотрим следующий пример матрицы 3x4, реализованной в виде списка из 3 списков длиной 4:
>>> matrix = [
... [1, 2, 3, 4],
... [5, 6, 7, 8],
... [9, 10, 11, 12],
... ]
Следующее списковое включение транспонирует строки и столбцы:
>>> [[row[i] for row in matrix] for i in range(4)]
[[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]
Как мы видели в предыдущем разделе, вложенное списочное выражение вычисляется в контексте for, которое следует за ним, поэтому этот пример эквивалентен:
>>> transposed = []
>>> for i in range(4):
... transposed.append([row[i] for row in matrix])
...
>>> transposed
[[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]
что, в свою очередь, то же самое, что:
>>> transposed = []
>>> for i in range(4):
... # следующие 3 строки реализуют вложенное списковое включение
... transposed_row = []
... for row in matrix:
... transposed_row.append(row[i])
... transposed.append(transposed_row)
...
>>> transposed
[[1, 5, 9], [2, 6, 10], [3, 7, 11], [4, 8, 12]]
На практике встроенные функции предпочтительнее сложных составных инструкций. Функция zip() отлично подходит для этого случая:
>>> list(zip(*matrix))
[(1, 5, 9), (2, 6, 10), (3, 7, 11), (4, 8, 12)]
Смотрите Распаковка списков аргументов для подробностей об звёздочке в этой строке.
5.2. Оператор del¶The del statement
Существует способ удалить элемент из списка по его индексу, а не по значению: оператор del. Он отличается от метода pop(), который возвращает значение. Оператор del также можно использовать для удаления срезов из списка или очистки всего списка (что мы делали ранее, присваивая пустой список срезу). Например:
>>> a = [-1, 1, 66.25, 333, 333, 1234.5]
>>> del a[0]
>>> a
[1, 66.25, 333, 333, 1234.5]
>>> del a[2:4]
>>> a
[1, 66.25, 1234.5]
>>> del a[:]
>>> a
[]
del также можно использовать для удаления целых переменных:
>>> del a
Ссылка на имя a в дальнейшем приведёт к ошибке (по крайней мере, пока ему не присвоено другое значение). Позже мы найдём другое применение для del.
5.3. Кортежи и последовательности¶Tuples and Sequences
Мы видели, что списки и строки имеют много общих свойств, таких как операции индексации и извлечения срезов. Это два примера последовательных типов данных (см. Типы последовательностей – list, tuple, range). Поскольку Python – развивающийся язык, могут быть добавлены другие типы последовательностей. Существует также ещё один стандартный тип последовательности: кортеж.
Кортеж состоит из нескольких значений, разделенных запятыми, например:
>>> t = 12345, 54321, 'hello!'
>>> t[0]
12345
>>> t
(12345, 54321, 'hello!')
>>> # Кортежи могут быть вложенными:
... u = t, (1, 2, 3, 4, 5)
>>> u
((12345, 54321, 'hello!'), (1, 2, 3, 4, 5))
>>> # Кортежи неизменяемы:
... t[0] = 88888
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: 'tuple' object does not support item assignment
>>> # но они могут содержать изменяемые объекты:
... v = ([1, 2, 3], [3, 2, 1])
>>> v
([1, 2, 3], [3, 2, 1])
Как видите, при выводе кортежи всегда заключаются в круглые скобки, чтобы вложенные кортежи интерпретировались правильно; их можно вводить как с окружающими скобками, так и без них, хотя скобки часто необходимы (если кортеж является частью более крупного выражения). Присваивать значения отдельным элементам кортежа нельзя, однако можно создавать кортежи, содержащие изменяемые объекты, например списки.
Хотя кортежи могут казаться похожими на списки, они часто используются в разных ситуациях и для разных целей. Кортежи являются неизменяемыми и обычно содержат разнородную последовательность элементов, доступ к которым осуществляется через распаковку (см. далее в этом разделе) или индексацию (или даже по атрибуту в случае именованных кортежей). Списки являются изменяемыми, и их элементы обычно однородны и доступны при переборе списка.
Особая проблема – создание кортежей, содержащих 0 или 1 элемент: в синтаксисе есть некоторые дополнительные особенности для таких случаев. Пустые кортежи создаются с помощью пустой пары круглых скобок; кортеж с одним элементом создается путем добавления запятой после значения (недостаточно заключить одно значение в скобки). Некрасиво, но эффективно. Например:
>>> empty = ()
>>> singleton = 'hello', # <-- обратите внимание на завершающую запятую
>>> len(empty)
0
>>> len(singleton)
1
>>> singleton
('hello',)
Оператор t = 12345, 54321, 'hello!' является примером упаковки кортежа: значения 12345, 54321 и 'hello!' упаковываются вместе в кортеж. Возможна и обратная операция:
>>> x, y, z = t
Это называется, что вполне уместно, распаковкой последовательности и работает для любой последовательности в правой части. Распаковка последовательности требует, чтобы переменных в левой части знака равенства было столько же, сколько элементов в последовательности. Обратите внимание, что множественное присваивание – это на самом деле комбинация упаковки кортежа и распаковки последовательности.
5.4. Множества¶Sets
Python также включает тип данных для множеств. Множество – это неупорядоченная коллекция без повторяющихся элементов. Основные применения: проверка принадлежности и удаление дубликатов. Объекты-множества также поддерживают математические операции, такие как объединение, пересечение, разность и симметрическая разность.
Фигурные скобки или функция set() могут использоваться для создания множеств. Примечание: для создания пустого множества нужно использовать set(), а не {}; последнее создаёт пустой словарь – структуру данных, которую мы обсудим в следующем разделе.
Вот краткая демонстрация:
>>> basket = {'apple', 'orange', 'apple', 'pear', 'orange', 'banana'}
>>> print(basket) # показать, что дубликаты удалены
{'orange', 'banana', 'pear', 'apple'}
>>> 'orange' in basket # быстрая проверка принадлежности
True
>>> 'crabgrass' in basket
False
>>> # Демонстрация операций над множествами с уникальными буквами из двух слов
...
>>> a = set('abracadabra')
>>> b = set('alacazam')
>>> a # уникальные буквы в a
{'a', 'r', 'b', 'c', 'd'}
>>> a - b # буквы, которые есть в a, но нет в b
{'r', 'd', 'b'}
>>> a | b # Буквы из a или b
{'a', 'c', 'r', 'd', 'b', 'm', 'z', 'l'}
>>> a & b # буквы, которые есть и в a, и в b
{'a', 'c'}
>>> a ^ b # буквы, которые есть в a или в b, но не в обоих
{'r', 'd', 'b', 'm', 'z', 'l'}
Аналогично списковым включениям, также поддерживаются множественные включения:
>>> a = {x for x in 'abracadabra' if x not in 'abc'}
>>> a
{'r', 'd'}
5.5. Словари¶Dictionaries
Ещё одним полезным типом данных, встроенным в Python, является словарь (см. Типы отображений – dict). В других языках словари иногда называются «ассоциативными запоминающими устройствами» или «ассоциативными массивами». В отличие от последовательностей, которые индексируются диапазоном чисел, словари индексируются ключами, которые могут быть любого неизменяемого типа; строки и числа всегда могут быть ключами. Кортежи могут использоваться как ключи, если они содержат только строки, числа или кортежи; если кортеж содержит любой изменяемый объект прямо или косвенно, он не может быть использован как ключ. Списки нельзя использовать как ключи, поскольку списки можно изменять на месте с помощью присваиваний по индексу, срезу или методов, таких как append() и extend().
Словарь лучше всего представлять как неупорядоченное множество пар ключ: значение, при этом ключи должны быть уникальными (в пределах одного словаря). Пара фигурных скобок создает пустой словарь: {}. Размещение внутри скобок списка пар ключ:значение, разделенных запятыми, добавляет начальные пары ключ:значение в словарь; так же словари отображаются при выводе.
Основные операции над словарём – это сохранение значения по некоторому ключу и извлечение значения по заданному ключу. Также можно удалить пару ключ:значение с помощью del. Если сохранить значение по уже используемому ключу, старое значение, связанное с этим ключом, будет забыто. Извлечение значения по несуществующему ключу является ошибкой.
Выполнение list(d.keys()) над словарём возвращает список всех ключей, используемых в словаре, в произвольном порядке (если требуется отсортированный список, используйте вместо этого sorted(d.keys())). [2] Чтобы проверить, находится ли отдельный ключ в словаре, используйте ключевое слово in.
Вот небольшой пример использования словаря:
>>> tel = {'jack': 4098, 'sape': 4139}
>>> tel['guido'] = 4127
>>> tel
{'sape': 4139, 'guido': 4127, 'jack': 4098}
>>> tel['jack']
4098
>>> del tel['sape']
>>> tel['irv'] = 4127
>>> tel
{'guido': 4127, 'irv': 4127, 'jack': 4098}
>>> list(tel.keys())
['irv', 'guido', 'jack']
>>> sorted(tel.keys())
['guido', 'irv', 'jack']
>>> 'guido' in tel
True
>>> 'jack' not in tel
False
Конструктор dict() создаёт словари непосредственно из последовательностей пар ключ-значение:
>>> dict([('sape', 4139), ('guido', 4127), ('jack', 4098)])
{'sape': 4139, 'jack': 4098, 'guido': 4127}
Кроме того, словарные включения можно использовать для создания словарей из произвольных выражений ключа и значения:
>>> {x: x**2 for x in (2, 4, 6)}
{2: 4, 4: 16, 6: 36}
Когда ключи являются простыми строками, иногда проще указать пары с помощью именованных аргументов:
>>> dict(sape=4139, guido=4127, jack=4098)
{'sape': 4139, 'jack': 4098, 'guido': 4127}
5.6. Техники циклов¶Looping Techniques
При переборе словарей ключ и соответствующее значение можно получить одновременно с помощью метода items().
>>> knights = {'gallahad': 'the pure', 'robin': 'the brave'}
>>> for k, v in knights.items():
... print(k, v)
...
gallahad the pure
robin the brave
При переборе последовательности индекс позиции и соответствующее значение можно получить одновременно с помощью функции enumerate().
>>> for i, v in enumerate(['tic', 'tac', 'toe']):
... print(i, v)
...
0 tic
1 tac
2 toe
Для одновременного перебора двух или более последовательностей элементы можно объединить в пары с помощью функции zip().
>>> questions = ['name', 'quest', 'favorite color']
>>> answers = ['lancelot', 'the holy grail', 'blue']
>>> for q, a in zip(questions, answers):
... print('What is your {0}? It is {1}.'.format(q, a))
...
What is your name? It is lancelot.
What is your quest? It is the holy grail.
What is your favorite color? It is blue.
Чтобы перебрать последовательность в обратном порядке, сначала укажите последовательность в прямом порядке, а затем вызовите функцию reversed().
>>> for i in reversed(range(1, 10, 2)):
... print(i)
...
9
7
5
3
1
Чтобы перебрать последовательность в отсортированном порядке, используйте функцию sorted(), которая возвращает новый отсортированный список, не изменяя исходный.
>>> basket = ['apple', 'orange', 'apple', 'pear', 'orange', 'banana']
>>> for f in sorted(set(basket)):
... print(f)
...
apple
banana
orange
pear
Чтобы изменить последовательность, по которой выполняется итерация внутри цикла (например, чтобы продублировать определённые элементы), рекомендуется сначала сделать её копию. Циклический перебор последовательности не создаёт копию неявно. Синтаксис среза делает это особенно удобным:
>>> words = ['cat', 'window', 'defenestrate']
>>> for w in words[:]: # Выполнить итерацию по копии среза всего списка.
... if len(w) > 6:
... words.insert(0, w)
...
>>> words
['defenestrate', 'cat', 'window', 'defenestrate']
5.7. Подробнее об условиях¶More on Conditions
Условия, используемые в операторах while и if, могут содержать любые операторы, а не только сравнения.
Операторы сравнения in и not in проверяют, встречается ли значение в последовательности (или не встречается). Операторы is и is not сравнивают, являются ли два объекта действительно одним и тем же объектом; это важно только для изменяемых объектов, таких как списки. Все операторы сравнения имеют одинаковый приоритет, который ниже приоритета всех числовых операторов.
Сравнения могут быть объединены в цепочку. Например, a < b == c проверяет, a меньше ли b и, более того, b равно c.
Сравнения могут объединяться с помощью логических операторов and и or, а результат сравнения (или любого другого логического выражения) может быть инвертирован с помощью not. Эти операторы имеют более низкий приоритет, чем операторы сравнения; среди них not имеет самый высокий приоритет, а or – самый низкий, так что A and not B or C эквивалентно (A and (not B)) or C. Как всегда, для выражения желаемой группировки можно использовать круглые скобки.
Логические операторы and и or являются так называемыми short-circuit операторами: их аргументы вычисляются слева направо, и вычисление прекращается, как только результат становится определённым. Например, если A и C истинны, а B ложно, то A and B and C не вычисляет выражение C. При использовании в качестве общего значения, а не логического, возвращаемым значением оператора короткого замыкания является последний вычисленный аргумент.
Результат сравнения или другого логического выражения можно присвоить переменной. Например,
>>> string1, string2, string3 = '', 'Trondheim', 'Hammer Dance'
>>> non_null = string1 or string2 or string3
>>> non_null
'Trondheim'
В Python, в отличие от C, присваивание не может использоваться внутри выражений. Программисты на C могут ворчать по этому поводу, но это позволяет избежать распространённого класса проблем, встречающихся в программах на C: ввод = в выражении, когда подразумевался ==.
5.8. Сравнение последовательностей и других типов¶Comparing Sequences and Other Types
Объекты последовательностей можно сравнивать с другими объектами того же типа последовательности. Сравнение использует лексикографический порядок: сначала сравниваются первые два элемента, и если они различаются, это определяет результат сравнения; если они равны, сравниваются следующие два элемента, и так до тех пор, пока одна из последовательностей не закончится. Если сравниваемые элементы сами являются последовательностями того же типа, лексикографическое сравнение выполняется рекурсивно. Если все элементы двух последовательностей равны, последовательности считаются равными. Если одна последовательность является начальной подпоследовательностью другой, то более короткая последовательность считается меньшей. Лексикографический порядок для строк использует номер кодовой точки Unicode для упорядочивания отдельных символов. Некоторые примеры сравнений между последовательностями одного типа:
(1, 2, 3) < (1, 2, 4)
[1, 2, 3] < [1, 2, 4]
'ABC' < 'C' < 'Pascal' < 'Python'
(1, 2, 3, 4) < (1, 2, 4)
(1, 2) < (1, 2, -1)
(1, 2, 3) == (1.0, 2.0, 3.0)
(1, 2, ('aa', 'ab')) < (1, 2, ('abc', 'a'), 4)
Обратите внимание, что сравнение объектов разных типов с помощью < или > допустимо при условии, что объекты имеют соответствующие методы сравнения. Например, смешанные числовые типы сравниваются по их числовому значению, поэтому 0 равно 0.0 и т.д. В противном случае, вместо того чтобы давать произвольный порядок, интерпретатор вызовет исключение TypeError.
Сноски
| [1] | В других языках может возвращаться изменённый объект, что позволяет выполнять цепочки методов, например d->insert("a")->remove("b")->sort();. |
| [2] | Вызов d.keys() вернёт объект представления словаря. Он поддерживает такие операции, как проверка принадлежности и итерация, но его содержимое не является независимым от исходного словаря – это всего лишь представление. |