Избавляемся от вложенных циклов с itertools.product

artem — Mon, 25 Nov 2019 08:23:23 +0000

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

Представим

colors = ['black', 'white', 'yellow']
animals = ['rabbit', 'shark', 'quokka']

Дефолт код, для прохода по всем комбинациям из массивов

for color in colors:
  for animal in animals:
    print(f'{color} {animal}')

сокращается с itertools

import itertools

for color, animal in itertools.product(colors, animals):
  print(f'{color} {animal}')

и как бонус однострочник

[print(f'{color} {animal}') for color, animal in itertools.product(colors, animals)]

как минус - создается новый объект, поэтому для больших объемов данных нецелесообразно
как плюс - в качестве аргументов можно передать любое количество массивов - получим все их комбинации

контекстные менеджеры

artem — Thu, 14 Nov 2019 12:51:31 +0000

https://docs.python.org/3/library/contextlib.html

в случае, когда у вас есть скоуп для применения ресурса, оч классно использовать контекстные менеджеры
объявляется декоратором @contextmanager, дальше при помощи with используется
вы наверняка встречали, когда работали с файлами:
with open('workfile') as f: позволяет вам провести все операции с файлом — он закроется автоматически при выходе из скоупа

пример кейса — коннекшн к бд, открываем его, в нем открываем курсор (контекстный менеджер в контекстном менеджере), в скоупе делаем что-то и конекшн закроется при выходе автоматом, и дальше его можно переиспользовать

объявление

@contextmanager
def connection(driver, server, database, user, password):
    params = f'DRIVER={{{driver}}};SERVER={server};DATABASE={database};' \
             f'Trusted_Connection=no;UID={user};PWD={password}'
    conn = pyodbc.connect(params)
    yield conn
    conn.close()


@contextmanager
def cursor(driver, server, database, user, password):
    with connection(driver, server, database, user, password) as conn:
        csr = conn.cursor()
        yield csr
        csr.close()

пример использования

        with cursor(driver=self.config.driver, server=self.config.address, database=database,
                    user=self.config.user, password=self.config.password) as curs:
            curs.execute(f'truncate table {table_name}')
            curs.commit()

или можно использовать конекшн, а не курсор напрямую (пандас работает через конекшн)

        with connection(driver=self.config.driver, server=self.config.address, database=database,
                        user=self.config.user, password=self.config.password) as conn:
            df = pandas.read_sql(script, conn)

DEV Community: artem

Избавляемся от вложенных циклов с itertools.product

контекстные менеджеры