Me gustaría generar una matriz completa de combinaciones de valores a partir de varias listas. Pero no siempre sé cuántas listas se proporcionarán y cada una puede tener una longitud diferente. Pude obtener la respuesta abusando de itertools.product (), pero creo que es excesivo. ¿Existe una forma más pitónica?
import itertools
listoflists = [
["happy","sad"],
["car","bike","truck"],
["true","false"],
[3,6,34,8,31]]
def comboGenerator(lol):
# create a uniform N-dimensional struct, big enough to hold longest dimension
longest = max(map(lambda x: len(x),lol))
superstruct = itertools.product(range(longest), repeat=len(lol))
# visit each cell in the struct, ignore the empty ones
for a in superstruct:
combo = []
try:
for index in range(len(lol)):
combo.append(lol[index][a[index]])
yield (combo)
except IndexError: #this was an empty cell; try again
pass
for x in comboGenerator(listoflists):
print (x)
result = list(itertools.product(*listoflists))
si desea que el tipo de elemento de la result
lista sea a list
, conviértalo de la siguiente manera:
result = [list(item) for item in result]
Este artículo se recopila de Internet, indique la fuente cuando se vuelva a imprimir.
En caso de infracción, por favor [email protected] Eliminar
Déjame decir algunas palabras