Estoy básicamente en busca de una versión de Python Combination of List<List<int>>Todas las combinaciones de una lista de listas
Dada una lista de listas, que necesitan una nueva lista que da todas las posibles combinaciones de elementos entre las listas.
[[1,2,3],[4,5,6],[7,8,9,10]] -> [[1,4,7],[1,4,8],...,[3,6,10]]
Se desconoce el número de listas, por lo que necesito algo que funcione para todos los casos. ¡Puntos de bonificación por elegancia!
necesita itertools.product:
>>> import itertools
>>> a = [[1,2,3],[4,5,6],[7,8,9,10]]
>>> list(itertools.product(*a))
[(1, 4, 7), (1, 4, 8), (1, 4, 9), (1, 4, 10), (1, 5, 7), (1, 5, 8), (1, 5, 9), (1, 5, 10), (1, 6, 7), (1, 6, 8), (1, 6, 9), (1, 6, 10), (2, 4, 7), (2, 4, 8), (2, 4, 9), (2, 4, 10), (2, 5, 7), (2, 5, 8), (2, 5, 9), (2, 5, 10), (2, 6, 7), (2, 6, 8), (2, 6, 9), (2, 6, 10), (3, 4, 7), (3, 4, 8), (3, 4, 9), (3, 4, 10), (3, 5, 7), (3, 5, 8), (3, 5, 9), (3, 5, 10), (3, 6, 7), (3, 6, 8), (3, 6, 9), (3, 6, 10)]
La solución más elegante es usar itertools.product en Python 2.6.
Si no está utilizando Python 2.6, la documentación para itertools.product muestran en realidad una función equivalente a hacer el producto de la manera "manual":
def product(*args, **kwds):
# product('ABCD', 'xy') --> Ax Ay Bx By Cx Cy Dx Dy
# product(range(2), repeat=3) --> 000 001 010 011 100 101 110 111
pools = map(tuple, args) * kwds.get('repeat', 1)
result = [[]]
for pool in pools:
result = [x+[y] for x in result for y in pool]
for prod in result:
yield tuple(prod)
estoy usando 2.6 ahora, pero gracias por la información! – Lin
listOLists = [[1,2,3],[4,5,6],[7,8,9,10]]
for list in itertools.product(*listOLists):
print list;
Espero que encuentre que a medida elegante como lo hice cuando lo encontré por primera vez.
¿Qué pasa con ese punto y coma? :) –
Fuerza del hábito. Me encanta cómo Python te permite poner un punto y coma, solo para ayudarnos a los programadores C/Java. Pero está claro; no es realmente un terminador de declaración cuando haces algo como print ("foo") ;; que es perfectamente legal en C o Java (aunque no tiene sentido) pero está prohibido en Python. –
Numpy puede hacerlo:
>>> import numpy
>>> a = [[1,2,3],[4,5,6],[7,8,9,10]]
>>> [list(x) for x in numpy.array(numpy.meshgrid(*a)).T.reshape(-1,len(a))]
[[ 1, 4, 7], [1, 5, 7], [1, 6, 7], ....]
¿Podría alguien explicar esto? – ashishv
No hay nada malo con la recursividad hacia arriba para esta tarea, y si necesita una versión que funciona con cadenas, esto podría satisfacer sus necesidades:
combinations = []
def combine(terms, accum):
last = (len(terms) == 1)
n = len(terms[0])
for i in range(n):
item = accum + terms[0][i]
if last:
combinations.append(item)
else:
combine(terms[1:], item)
>>> a = [['ab','cd','ef'],['12','34','56']]
>>> combine(a, '')
>>> print(combinations)
['ab12', 'ab34', 'ab56', 'cd12', 'cd34', 'cd56', 'ef12', 'ef34', 'ef56']
Podría alguien explica el significado del asterisco en '* a'? – Serrano
'* a' significa que estos son argumentos que se pasan a la función o método. 'def fn (a, b, c):' respondería a 'fn (* [1,2,3])' [reference] (http://www.saltycrane.com/blog/2008/01/how- a-uso-args-y-kwargs-en-python /) – mjallday
@mjallday, sería posible añadir también estas combinaciones: (7,4,1), (8,4,1), (9,4, 1), (10,4,1), (7,5,1), (8,5,1), (9,5,1), (10,5,1), etc. – Reman