Как развернуть содержимое args?

Question

[nihi1ist]

Пытаюсь решить задачу:

Напишите функцию, которая будет являться аналогом функции zip и сделайте так, чтобы программа работала с любыми итерируемыми типами данных. Циклами (за исключением генераторов), условными операторами и функциями определения типа(isinstance или type) пользоваться нельзя. Функция должна возвращать список из пар кортежей.

Решал похожую задачу, но там была пара коллекций. А тут как мне объяснили, нужно сделать как в оригинале, чтобы можно было обрабатывать произвольное количество коллекций. И вот никак в голове не формируется алгоритм такой функции. Понимаю, что нужно использовать *args. Дальше хочу уже list comprehension прикрутить, но за несколько дней ничего не вышло.

Answer 1

[LXSTVAYNE]

Можно как-то так сделать:

def my_zip(*iterables):
    min_len = min(map(len, iterables))
    iterators = [iter(iterable) for iterable in iterables]
    return [tuple(map(next, iterators))
            for _ in range(min_len)]

Comments

[nihi1ist]

LXSTVAYNE надо отдельно про map прочитать, моё решение получилось более громоздким и менее изящным.

[LXSTVAYNE]

nihi1ist, в программировании можно улучшать навыки очень долго, я вот даже не уверен, что мой вариант самый изящный)

Answer 2

[Ivan Yakushenko]

from typing import List, Iterable, Any


def super_zip(*args: Iterable) -> List[tuple]:
    count = len(args)
    lengths = {i: len(v) for i, v in enumerate(args)}
    max_values_index = max(lengths, key=lengths.get)

    def extract(chunk_index: int, value_index: int) -> Any:
        try:
            return args[chunk_index][value_index]
        except IndexError:
            return None

    return [
        tuple(extract(ci, vi) for ci in range(count))
            for vi in range(lengths[max_values_index])
    ]

super_zip('abc', [1, 2, 3], ('Z', 'X', 'C'))
# [('a', 1, 'Z'), ('b', 2, 'X'), ('c', 3, 'C')]

super_zip('abc', [1, 2, 3, 4], ('Z', 'X'))
# [('a', 1, 'Z'), ('b', 2, 'X'), ('c', 3, None), (None, 4, None)]

super_zip('abc')
# [('a',), ('b',), ('c',)]

Comments

[nihi1ist]

Ivan Yakushenko написал свой вариант. Но есть проблема, если формирую список со вложенными кортежами, то результат кривой. А если список со вложенными списками, то нормально. Не подскажите почему?

spoiler

def zip_super(*args):
    args = [list(item) for item in args]
    min_col = len(min(args, key=len))
    col_counter = len(args)
    temp = [args[number][index] for index in range(min_col) for number in range(col_counter)]
    result = [[temp[index + step] for index in range(col_counter)] for step in range(0, len(temp), col_counter)]
    return result


zipped = zip_super('asdf', [1, 2, 3], 'lkjhg', {5, 6, 7, 8}, {'a': 0, 'b': 1, 'c': 2})
# print(zipped)
for z in zipped:
    print(z)

[nihi1ist]

Решил задачу. Вместо круглых скобок, метод tuple() использовал. И с множествами со словарями разобрался. Правда вот по этому участку есть сомнения и желание уместить в один list comprehension:

temp = [args[number][index] for index in range(min_col) for number in range(col_counter)]
result = [[temp[index + step] for index in range(col_counter)] for step in range(0, len(temp), col_counter)]

Если подскажете чего, было бы замечательно.

[Ivan Yakushenko]

nihi1ist, зачем так усложнять и делать лишние итерации? Разберись с основной логикой в каком порядке необходимо делать выборку. В моем примере достаточно изменить 1 строчку кода и итерировать не до максимума, а до минимума значений в чанке и тогда результат будет в точности, как у zip

def super_zip(*args: Iterable) -> List[tuple]:
    count = len(args)
    lengths = {i: len(v) for i, v in enumerate(args)}
    min_values_index = min(lengths, key=lengths.get)

    return [
        tuple(args[ci][vi] for ci in range(count))
            for vi in range(lengths[min_values_index])
    ]

List comprehensions нужны, что бы упростить конструкции, у тебя же и близко нет упрощения. В конечном итоге количество кода у тебя больше, чем у меня и его читаемость гораздо хуже. Просто потому что за точку отсчета взял не тот параметр.

[LXSTVAYNE]

Ivan Yakushenko, try...except в алгоритмах использовать же не лучшая идея ? Или python оптимизирует как-то возбуждение эксепшена?

[nihi1ist]

Ivan Yakushenko, с множествами не работает. Хоть и громоздко получилось у меня, но код работает.