10

各内部リストの長さが1またはn（n> 1と仮定）になるようなリストのリストがあります。

>>> uneven = [[1], [47, 17, 2, 3], [3], [12, 5, 75, 33]]

リストを転置したいのですが、長いリストを（のようにzip）切り捨てたり、短いリストをNoneで埋めたりする代わりに、短いリストを独自の特異値で埋めたいと思います。言い換えれば、私は取得したいです：

>>> [(1, 47, 3, 12), (1, 17, 3, 5), (1, 2, 3, 75), (1, 3, 3, 33)]

これは、2、3回繰り返すことで実行できます。

>>> maxlist = len(max(*uneven, key=len))
>>> maxlist
4
>>> from itertools import repeat
>>> uneven2 = [x if len(x) == maxlist else repeat(x[0], maxlist) for x in uneven]
>>> uneven2
[[1, 1, 1, 1], [47, 17, 2, 3], [3, 3, 3, 3], [12, 5, 75, 33]]
>>> zip(*uneven2)
[(1, 47, 3, 12), (1, 17, 3, 5), (1, 2, 3, 75), (1, 3, 3, 33)]

しかし、より良いアプローチはありますか？これを達成するために、私は本当にmaxlist事前に知る必要がありますか？

4

3 に答える 3

7

1つの要素リストを永久に繰り返すことができます。

uneven = [[1], [47, 17, 2, 3], [3], [12, 5, 75, 33]]

from itertools import repeat

print zip(*(repeat(*x) if len(x)==1 else x for x in uneven))
于 2012-05-16T00:29:08.230 に答える
4

代わりに使用できますitertools.cycle()：

>>> from itertools import cycle
>>> uneven3 = [x if len(x) != 1 else cycle(x) for x in uneven]
>>> zip(*uneven3)
[(1, 47, 3, 12), (1, 17, 3, 5), (1, 2, 3, 75), (1, 3, 3, 33)]

つまり、maxlist事前に知る必要はありません。

于 2012-05-16T00:30:14.927 に答える
0

@ chris-morganのシミュレーションのアイデアが本当に好きだっitertools.izip_longestたので、ようやくインスピレーションを得たときにizip_cycle関数を作成しました。

def izip_cycle(*iterables, **kwargs):
    """Make an iterator that aggregates elements from each of the iterables.
    If the iterables are of uneven length, missing values are filled-in by cycling the shorter iterables.
    If an iterable is empty, missing values are fillvalue or None if not specified.
    Iteration continues until the longest iterable is exhausted.
    """
    fillvalue = kwargs.get('fillvalue')
    counter = [len(iterables)]
    def cyclemost(iterable):
        """Cycle the given iterable like itertools.cycle, unless the counter has run out."""
        itb = iter(iterable)
        saved = []
        try:
            while True:
                element = itb.next()
                yield element
                saved.append(element)
        except StopIteration:
            counter[0] -= 1
            if counter[0] > 0:
                saved = saved or [fillvalue]
                while saved:
                    for element in saved:
                        yield element
    iterators = [cyclemost(iterable) for iterable in iterables]
    while iterators:
        yield tuple([next(iterator) for iterator in iterators])

print list(izip_cycle([], range(3), range(6), fillvalue='@'))
于 2012-06-25T12:29:52.273 に答える