Наївний підхід
def transpose_finite_iterable(iterable):
return zip(*iterable) # `itertools.izip` for Python 2 users
прекрасно працює для кінцево ітерабельних (наприклад, послідовностей, як list
/ tuple
/ str
) (потенційно нескінченних) ітерабелів, які можна проілюструвати як
| |a_00| |a_10| ... |a_n0| |
| |a_01| |a_11| ... |a_n1| |
| |... | |... | ... |... | |
| |a_0i| |a_1i| ... |a_ni| |
| |... | |... | ... |... | |
де
n in ℕ
,
a_ij
відповідає j
-му елементу i
-й ітерабельний,
і після застосування transpose_finite_iterable
ми отримуємо
| |a_00| |a_01| ... |a_0i| ... |
| |a_10| |a_11| ... |a_1i| ... |
| |... | |... | ... |... | ... |
| |a_n0| |a_n1| ... |a_ni| ... |
Приклад Python такого випадку, коли a_ij == j
,n == 2
>>> from itertools import count
>>> iterable = [count(), count()]
>>> result = transpose_finite_iterable(iterable)
>>> next(result)
(0, 0)
>>> next(result)
(1, 1)
Але ми не можемо використати transpose_finite_iterable
знову, щоб повернутися до структури оригіналу, iterable
оскільки result
це нескінченний ітерабел кінцевих ітерабелів ( tuple
у нашому випадку):
>>> transpose_finite_iterable(result)
... hangs ...
Traceback (most recent call last):
File "...", line 1, in ...
File "...", line 2, in transpose_finite_iterable
MemoryError
То як ми можемо розібратися з цією справою?
... і ось ось deque
Після того, як ми розглянемо документи itertools.tee
функції , є рецепт Python, який з деякою модифікацією може допомогти в нашому випадку
def transpose_finite_iterables(iterable):
iterator = iter(iterable)
try:
first_elements = next(iterator)
except StopIteration:
return ()
queues = [deque([element])
for element in first_elements]
def coordinate(queue):
while True:
if not queue:
try:
elements = next(iterator)
except StopIteration:
return
for sub_queue, element in zip(queues, elements):
sub_queue.append(element)
yield queue.popleft()
return tuple(map(coordinate, queues))
давайте перевіримо
>>> from itertools import count
>>> iterable = [count(), count()]
>>> result = transpose_finite_iterables(transpose_finite_iterable(iterable))
>>> result
(<generator object transpose_finite_iterables.<locals>.coordinate at ...>, <generator object transpose_finite_iterables.<locals>.coordinate at ...>)
>>> next(result[0])
0
>>> next(result[0])
1
Синтез
Тепер ми можемо визначити загальну функцію для роботи з ітерабелями ітерабелів, один з яких є кінцевим, а інший - потенційно нескінченним, використовуючи functools.singledispatch
декоратор, як
from collections import (abc,
deque)
from functools import singledispatch
@singledispatch
def transpose(object_):
"""
Transposes given object.
"""
raise TypeError('Unsupported object type: {type}.'
.format(type=type))
@transpose.register(abc.Iterable)
def transpose_finite_iterables(object_):
"""
Transposes given iterable of finite iterables.
"""
iterator = iter(object_)
try:
first_elements = next(iterator)
except StopIteration:
return ()
queues = [deque([element])
for element in first_elements]
def coordinate(queue):
while True:
if not queue:
try:
elements = next(iterator)
except StopIteration:
return
for sub_queue, element in zip(queues, elements):
sub_queue.append(element)
yield queue.popleft()
return tuple(map(coordinate, queues))
def transpose_finite_iterable(object_):
"""
Transposes given finite iterable of iterables.
"""
yield from zip(*object_)
try:
transpose.register(abc.Collection, transpose_finite_iterable)
except AttributeError:
# Python3.5-
transpose.register(abc.Mapping, transpose_finite_iterable)
transpose.register(abc.Sequence, transpose_finite_iterable)
transpose.register(abc.Set, transpose_finite_iterable)
що може розглядатися як власна зворотна (математики називають цей вид функцій "інволюціями" ) в класі бінарних операторів над кінцевими не порожніми ітерабелями.
Як бонус singledispatch
ing, ми можемо обробляти numpy
масиви на зразок
import numpy as np
...
transpose.register(np.ndarray, np.transpose)
а потім використовувати його як
>>> array = np.arange(4).reshape((2,2))
>>> array
array([[0, 1],
[2, 3]])
>>> transpose(array)
array([[0, 2],
[1, 3]])
Примітка
Оскільки transpose
повертає ітератори , і якщо хто - то хоче мати tuple
в list
с , як в OP - це може бути зроблено додатково з map
вбудованою функцією , як
>>> original = [('a', 1), ('b', 2), ('c', 3), ('d', 4)]
>>> tuple(map(list, transpose(original)))
(['a', 'b', 'c', 'd'], [1, 2, 3, 4])
Реклама
Я додав узагальнене рішення до lz
пакета з 0.5.0
версії, яку можна використовувати як
>>> from lz.transposition import transpose
>>> list(map(tuple, transpose(zip(range(10), range(10, 20)))))
[(0, 1, 2, 3, 4, 5, 6, 7, 8, 9), (10, 11, 12, 13, 14, 15, 16, 17, 18, 19)]
PS
Не існує рішення (принаймні очевидного) для поводження з потенційно нескінченним ітерабелом потенційно нескінченних ітерабелів, але цей випадок зустрічається рідше.