itertooldict 0.1.2

A missing feature of itertools: labeled product of multiple choices from a dictionary.

itertooldict

A missing feature of itertools: labeled product of multiple choices from a dictionary.

Installation

pip install itertooldict

Usage

itertooldict takes a dictionary where values are iterables and yields dictionaries representing their Cartesian product.

from itertooldict import itertooldict

data = {
    "voltage": ["Vmax", "Vmin"],
    "temp": ["hot", "cold"]
}

for combo in itertooldict(data):
    print(combo)

# Output:
# {'voltage': 'Vmax', 'temp': 'hot'}
# {'voltage': 'Vmax', 'temp': 'cold'}
# {'voltage': 'Vmin', 'temp': 'hot'}
# {'voltage': 'Vmin', 'temp': 'cold'}

Excluding Combinations

You can use the .remove() method to exclude specific combinations or patterns.

it = itertooldict(data)
it.remove({"voltage": "Vmax", "temp": "hot"})

for combo in it:
    print(combo)
# {'voltage': 'Vmax', 'temp': 'cold'}
# {'voltage': 'Vmin', 'temp': 'hot'}
# {'voltage': 'Vmin', 'temp': 'cold'}

You can also remove based on partial matches:

Randomizing Order

You can randomize the order of the generated combinations:

it = itertooldict(data).random()
for combo in it:
    print(combo) # Order will be shuffled

Updating Key Order

You can specify the order of keys in the resulting dictionaries:

it = itertooldict(data)
it.updateKeyOrder(["temp", "voltage"])
for combo in it:
    print(combo) # {'temp': 'hot', 'voltage': 'Vmax'}, ...

Compatibility with list() and enumerate()

itertooldict works seamlessly with standard Python functions:

# Convert to list
all_combos = list(itertooldict(data))

# Use with enumerate
for i, combo in enumerate(itertooldict(data)):
    print(f"{i}: {combo}")