Tengo un diccionario con valores en ints, bools, y cadenas y listas y una lista ordenada de claves en la que quiero la lista final de listas. Quiero la salida de modo que donde haya una lista, hagamos combinaciones de eso con valores únicos.

Entrada:

cols = ['region', 'city', 'country', 'valid'] 
vals = {'city': [10, 20], 'valid': True, 'region': 3, 'country': 'US'}

Salida:

[[3, 10, 'US', True], [3, 20, 'US', True]]

Ahora, si otro campo es una lista, aumentaría las combinaciones.

Entrada:

cols = ['region', 'city', 'country', 'valid'] 
vals = {'city': [10, 20], 'valid': True, 'region': [3, 4], 'country': 'US'}

Salida:

[[3, 10, 'US', True], [3, 20, 'US', True], [4, 10, 'US', True], [4, 20, 'US', True]]

Buscando la forma más pitónica de lograr esto.

python
0
umarr 9 oct. 2019 a las 11:47

3 respuestas

La mejor respuesta

Utilice itertools.product:

import itertools

cols = ['region', 'city', 'country', 'valid'] 
vals = {'city': [10, 20], 'valid': True, 'region': [3, 4], 'country': 'US'}

values = [vals[key] for key in cols]
values = [val if isinstance(val, list) else [val] for val in values]

result = list(itertools.product(*values))
print(result)
# [(3, 10, 'US', True), (3, 20, 'US', True), (4, 10, 'US', True), (4, 20, 'US', True)]

Lectura relacionada:

3
Aran-Fey 9 oct. 2019 a las 10:01

Puede usar itertools.product para los campos respectivos como,

Pero antes de hacer product, debe verificar si es una instancia de int o str, en caso afirmativo, luego conviértalos en list o {{X4} } por el bien de la iteración correcta

>>> vals = {'city': [10, 20], 'valid': True, 'region': 3, 'country': 'US'}
>>> 
>>> city = vals['city']
>>> region = vals['region']
>>> country = vals['country']
>>> 
>>> if not isinstance(region, (list, tuple)):
...   region = [region]
... 
>>> if not isinstance(country, (list, tuple)):
...   country = [country]
... 
>>> list(product(region, city, country, [vals['valid']]))
[(3, 10, 'US', True), (3, 20, 'US', True)]


>>> vals
{'city': [10, 20], 'valid': True, 'region': [3, 4], 'country': 'US'}
>>> from itertools import product
>>> list(product(vals['region'], vals['city'], [vals['country']], [vals['valid']]))
[(3, 10, 'US', True), (3, 20, 'US', True), (4, 10, 'US', True), (4, 20, 'US', True)]

Podrías escribir una función para manejar eso como,

$ cat mkcomb.py
from itertools import product


def mk_comb(keys, vals):
    values = [
      vals[key] if isinstance(vals[key], (list, tuple)) else [vals[key]]
      for key in keys
    ]
    return list(product(*values))

cols = ['region', 'city', 'country', 'valid'] 
vals1 = {'city': [10, 20], 'valid': True, 'region': 3, 'country': 'US'}
vals2 = {'city': [10, 20], 'valid': True, 'region': [3, 4], 'country': 'US'}
print(mk_comb(cols, vals1))
print(mk_comb(cols, vals2))

Salida:

$ python mkcomb.py
[(3, 10, 'US', True), (3, 20, 'US', True)]
[(3, 10, 'US', True), (3, 20, 'US', True), (4, 10, 'US', True), (4, 20, 'US', True)]
2
han solo 9 oct. 2019 a las 09:20

La comprensión de la lista puede hacer el trabajo por usted:

list(itertools.product(*[vals[c] if type(vals[c]) == list else [vals[c]] for c in cols]))
0
tituszban 9 oct. 2019 a las 08:57
58300302

Preguntas relacionadas

30 ¿Cómo fusiono dos diccionarios en una sola expresión?
20 Convertir cadena en fecha y hora
19 Convertir bytes a una cadena
29 ¿Cómo hacer una lista plana de la lista de listas?
17 ¿Cómo puedo agregar nuevas claves a un diccionario?
10 Iterando sobre diccionarios usando bucles 'for'

Nuevas preguntas

3 Estoy dentro del rango usando pop pero todavía obtengo un error de fuera de rango
4 Use la variable de clase dentro de un método de esta clase
2 Clase de Python con muchas variables
3 Fusionando dos df sin ninguna clave
3 ¿Cómo acceder al valor en una cadena JSON usando una clave en Python?
3 convertir un entero de cadena con comas y punto para flotar
2 (Sintaxis) Obtener el índice de la mayor diferencia entre dos listas (en 1 línea)
2 Pandas: toma todos los caracteres antes de que aparezca el primer dígito
2 El acceso al diccionario de Python termina como un error de tipo
2 Pandas DataFrame: filtrar columna / índice por tipo
Nuevas preguntas con la etiqueta python >

python

Python es un lenguaje de programación multipropósito, de tipificación dinámica y de múltiples paradigmas. Está diseñado para ser rápido de aprender, comprender y usar, y hacer cumplir una sintaxis limpia y uniforme. Tenga en cuenta que Python 2 está oficialmente fuera de soporte a partir del 01-01-2020. Aún así, para preguntas de Python específicas de la versión, agregue la etiqueta [python-2.7] o [python-3.x]. Cuando utilice una variante de Python (por ejemplo, Jython, PyPy) o una biblioteca (por ejemplo, Pandas y NumPy), inclúyala en las etiquetas.

Más sobre python...