Число элементов в коллекциях Python

Расширенные примеры подсчёта элементов

Далее приведены менее распространённые, но полезные сценарии определения количества элементов в Python.

1. Подсчёт элементов в итераторе с помощью reduce

Функция functools.reduce позволяет свернуть итератор в одно значение. Для подсчёта используется единичное приращение на каждом шаге.

from functools import reduce

iterator = iter(range(1, 1001))
count = reduce(lambda acc, _: acc + 1, iterator, 0)
print(count)  # 1000

1000

Пояснение: reduce последовательно применяет функцию к аккумулятору (начинается с 0) и каждому элементу. Подходит, если требуется совместить подсчёт с другой операцией, например фильтрацией.

2. Подсчёт уникальных элементов через преобразование в множество

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

data = [1, 2, 2, 3, 4, 4, 4, 5]
unique_count = len(set(data))
print(unique_count)  # 5

5

Пояснение: множество автоматически удаляет дубликаты. Этот способ не изменяет исходную коллекцию и работает за O(n). Однако требует хранения всех уникальных элементов в памяти.

3. Подсчёт элементов в мультисловаре (словарь списков значений)

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

multi_dict = {
    'fruits': ['apple', 'banana'],
    'veggies': ['carrot', 'broccoli', 'pepper'],
    'grains': ['rice', 'pasta']
}
total = sum(len(v) for v in multi_dict.values())
print(total)  # 7

7

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

4. Подсчёт количества файлов в директории с помощью os.listdir

Модуль os предоставляет доступ к файловой системе. Для подсчёта файлов (без учёта поддиректорий) достаточно получить список содержимого.

import os

path = '.'
count_files = len([name for name in os.listdir(path) if os.path.isfile(os.path.join(path, name))])
print(f'Файлов в текущей папке: {count_files}')

Файлов в текущей папке: 15

Пояснение: os.listdir возвращает все имена в папке. Фильтрация по isfile оставляет только файлы. Если нужно считать и папки, условие меняется.

5. Подсчёт перекрывающихся вхождений подстроки в строке

Стандартный str.count не учитывает перекрытия. Собственная функция решает эту задачу.

def overlapping_count(text, sub):
    count = start = 0
    while True:
        pos = text.find(sub, start)
        if pos == -1:
            break
        count += 1
        start = pos + 1  # сдвиг на один символ для перекрытия
    return count

print(overlapping_count('aaa', 'aa'))  # 2
print(overlapping_count('ababa', 'aba'))  # 2

2
2

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

6. Приблизительная длина итератора с помощью operator.length_hint

Функция operator.length_hint возвращает предполагаемую длину итератора, если она известна (например, для range). Не гарантирует точности.

from operator import length_hint

it = iter(range(1, 100))
hint = length_hint(it)
print(hint)  # 99 (точное значение)

# Для генератора с неизвестной длиной
gen = (x for x in range(5))
hint2 = length_hint(gen)
print(hint2)  # 0 (неизвестно)

99
0

Пояснение: length_hint полезен для оптимизации аллокаций, когда точная длина не критична. Для пользовательских итераторов можно реализовать __length_hint__.

7. Итеративный подсчёт в глубоких древовидных структурах

Вместо рекурсии можно использовать явный стек, чтобы избежать переполнения стека при большой глубине.

def iterative_deep_count(root):
    stack = [root]
    count = 0
    while stack:
        obj = stack.pop()
        if isinstance(obj, (list, tuple, set)):
            stack.extend(obj)
        elif isinstance(obj, dict):
            for k, v in obj.items():
                stack.append(k)
                stack.append(v)
        else:
            count += 1
    return count

deep_list = [1, [2, [3, [4, 5]]]]
print(iterative_deep_count(deep_list))  # 5

5

Пояснение: стек имитирует рекурсивный обход. Порядок обработки (снизу вверх) может отличаться, но количество элементов остаётся одинаковым. При наличии циклических ссылок необходимо добавить множество посещённых объектов.