Как получить следующее число в Python? Полное руководство

Расширенные примеры и нестандартные сценарии

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

1. Инкремент чисел с плавающей точкой и комплексных чисел

Оператор += работает не только с int, но и с float, complex, Decimal и другими числовыми типами.

f = 3.14
f += 1.5
print(f)  # 4.64

c = 2+3j
c += 1
print(c)  # (3+3j)

4.64
(3+3j)

2. Пользовательский итератор с определением __next__

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

class SquareNumbers:
    def __init__(self, max_n):
        self.n = 0
        self.max = max_n
    def __iter__(self):
        return self
    def __next__(self):
        if self.n >= self.max:
            raise StopIteration
        result = self.n ** 2
        self.n += 1
        return result

squares = SquareNumbers(5)
for sq in squares:
    print(sq)

0
1
4
9
16

3. Использование lambda и map для создания списка следующих чисел

Функция map может применить операцию +1 ко всем элементам последовательности.

nums = [10, 20, 30]
next_nums = list(map(lambda x: x + 1, nums))
print(next_nums)

[11, 21, 31]

4. Обработка StopIteration с заданным значением по умолчанию

Второй аргумент next() возвращается, если итератор исчерпан, вместо исключения.

items = iter([1, 2, 3])
print(next(items, 'конец'))
print(next(items, 'конец'))
print(next(items, 'конец'))
print(next(items, 'конец'))

1
2
3
конец

5. Генерация следующего числа с произвольным шагом (числа Фибоначчи)

Генератор может реализовать сложные последовательности, например, ряд Фибоначчи.

def fibonacci():
    a, b = 0, 1
    while True:
        yield a
        a, b = b, a + b

fib = fibonacci()
for _ in range(7):
    print(next(fib))

0
1
1
2
3
5
8

6. Использование itertools.islice для получения нескольких следующих чисел

islice позволяет взять заданное количество элементов из бесконечного итератора.

from itertools import count, islice
counter = count(1, 0.5)
first_five = list(islice(counter, 5))
print(first_five)

[1, 1.5, 2.0, 2.5, 3.0]

7. Применение functools.reduce для последовательного инкремента

reduce может применить операцию сложения к накопленному значению, но это скорее демонстрация, не рекомендуется для простого инкремента.

from functools import reduce
nums = [1, 2, 3, 4]
result = reduce(lambda acc, x: acc + x, nums, 0)
print(result)  # 10

10

8. Работа с Decimal для точного финансового инкремента

Тип Decimal из модуля decimal обеспечивает точность, необходимую для денежных расчетов.

from decimal import Decimal
price = Decimal('19.99')
price += Decimal('0.01')
print(price)

20.00

9. Использование yield from для делегирования генерации

Конструкция yield from позволяет одному генератору передавать управление другому.

def sub_counter(start, end):
    for n in range(start, end):
        yield n

def main_counter():
    yield from sub_counter(1, 4)
    yield from sub_counter(10, 13)

for num in main_counter():
    print(num)

1
2
3
10
11
12

10. Применение operator.add для инкремента в функциональном стиле

Модуль operator содержит функции-эквиваленты операторов, например add.

from operator import add
x = 5
x = add(x, 1)
print(x)

6