Das kartesische Produkt von a dictionary of Listen

Ich versuche, code zu schreiben, zu testen, das kartesische Produkt von einer Reihe von Eingangsparametern.

Habe ich mir angeschaut itertools, aber seine product Funktion ist nicht genau das, was ich will. Gibt es einen einfachen, offensichtlichen Weg, um ein Wörterbuch mit einer beliebigen Anzahl von Schlüssel und eine beliebige Anzahl von Elementen in jedem Wert, und dann liefern Sie ein Wörterbuch mit der nächsten permutation?

Eingang:

options = {"number": [1,2,3], "color": ["orange","blue"] }
print list( my_product(options) )

Beispiel-Ausgabe:

[ {"number": 1, "color": "orange"},
  {"number": 1, "color": "blue"},
  {"number": 2, "color": "orange"},
  {"number": 2, "color": "blue"},
  {"number": 3, "color": "orange"},
  {"number": 3, "color": "blue"}
]
  • Ich bin mir ziemlich sicher, Sie brauchen keine Bibliothek, dies zu tun, aber ich weiß nicht, Python ganz gut genug zu beantworten. Ich denke, die Liste Verstehens sind der trick.
  • Ich Frage, ob gibt es eine ready-made-generator, der leicht angepasst werden kann, so etwas zu tun. Liste Verstehens sind überhaupt nicht relevant.
InformationsquelleAutor Seth Johnson | 2011-03-08
  1. 48

    Ok, danke @dfan für mich zu sagen, ich war an der falschen Stelle suchen. Ich habe es jetzt:

    from itertools import product
def my_product(inp):
    return (dict(zip(inp.keys(), values)) for values in product(*inp.values())

    BEARBEITEN: nach Jahren mehr Python Erfahrung, ich denke, eine bessere Lösung zu akzeptieren kwargs eher als ein Wörterbuch der Eingänge; der Aufruf Arbeitsweise ist Analog zu der des ursprünglichen itertools.product. Auch ich denke, dass das schreiben einer generator-Funktion, anstatt eine Funktion zurückgibt, die einen generator-Ausdruck, lässt den code übersichtlicher. Also:

    def product_dict(**kwargs):
    keys = kwargs.keys()
    vals = kwargs.values()
    for instance in itertools.product(*vals):
        yield dict(zip(keys, instance))

    und wenn Sie brauchen, um passieren in einem dict, list(product_dict(**mydict)). Die einzige Bemerkenswerte änderung mit kwargs eher als eine beliebige Eingabe-Klasse ist, dass es verhindert, dass die Schlüssel/Werte aus bestellt, zumindest bis Python 3.6.

    • Bedeutet die Tatsache, dass die Wörterbuch-Einträge werden ungeordnete, beeinflussen diese in sowieso?
    • Dies ist eine sehr nette-code zum schnellen generieren von unit-Testcases (cross-validation-set-Stil!)
    • Für Python 3 Benutzer. Ich habe eine aktualisierte version hier
    • Ich würde sagen, es ist okay, als Schlüssel und Werte, auch wenn ungeordnet, sind noch immer bestellt bzw. auf der jeweils anderen.
    • Wenn Sie mit dieser Liste von dictionaries eine Liste von **kwargs zu senden, um eine Funktion, über map ist, dann ist es ähnlich wie eine Menge von geschachtelten for-Schleifen. Der Unterschied ist, dass Sie keine Garantie für die Schleife auf der Außenseite, und die loop ist auf der Innenseite.
    InformationsquelleAutor Seth Johnson
  2. 16

    Python-3-version von Seths Antwort.

    import itertools

def dict_product(dicts):
    """
    >>> list(dict_product(dict(number=[1,2], character='ab')))
    [{'character': 'a', 'number': 1},
     {'character': 'a', 'number': 2},
     {'character': 'b', 'number': 1},
     {'character': 'b', 'number': 2}]
    """
    return (dict(zip(dicts, x)) for x in itertools.product(*dicts.values()))
    InformationsquelleAutor Tarrasch
  3. 6

    Durch die Art und Weise, dies ist nicht eine permutation. Eine permutation ist eine Neuordnung der Liste. Dies ist eine Aufzählung der möglichen Auswahl aus Listen.

    Edit: nach erinnern, dass es als ein Kartesisches Produkt, ich kam mit dieser:

    import itertools
options = {"number": [1,2,3], "color": ["orange","blue"] }
product = [x for x in apply(itertools.product, options.values())]
print [dict(zip(options.keys(), p)) for p in product]
    • Ich habe versucht zu erklären, warum suchen "Permutationen" war nicht zu helfen. Ich erinnerte mich, was das eigentlich ist: es ist ein Kartesisches Produkt. Ich würde beginnen Sie mit itertools.Produkt().
    • Yep, wird gemacht, und danke für den pointer. Aber trotzdem, herzlich willkommen auf Stack Overflow: eine Antwort sollten Sie eine, die bietet tatsächlich eine Antwort auf die Frage. Das gehört als Kommentar auf die Frage.
    • nicht wirklich, die in der Originalversion nicht Zustand Kartesisches Produkt
    • Ich scheine nicht die Möglichkeit haben zu kommentieren auf nichts, aber meine eigenen Antworten noch. Ich würde es wenn ich könnte. Ich bin froh, dass meine Antwort führte Sie die Lösung.
    • Ah, verstanden. Gut, vielen Dank nochmal für deine Hilfe beim setzen mich auf die richtige Spur.
    InformationsquelleAutor dfan
  4. 2
    # I would like to do
keys,values = options.keys(), options.values()
# but I am not sure that the keys and values would always
# be returned in the same relative order. Comments?
keys = []
values = []
for k,v in options.iteritems():
    keys.append(k)
    values.append(v)

import itertools
opts = [dict(zip(keys,items)) for items in itertools.product(*values)]

    Ergebnisse in

    opts = [
    {'color': 'orange', 'number': 1},
    {'color': 'orange', 'number': 2},
    {'color': 'orange', 'number': 3},
    {'color': 'blue', 'number': 1},
    {'color': 'blue', 'number': 2},
    {'color': 'blue', 'number': 3}
]
    • Ich denke Python garantiert, dass die keys() und values() und den entsprechenden iter - * zurück in der gleichen Reihenfolge. Siehe docs.python.org/library/stdtypes.html#dict.items
    • Ausgezeichnet! Danke, das hatte wurde stört mich für eine Weile.
    • Sie sind Recht herzlich willkommen. Es ist sehr handlich und besonders für diesen Fall. Wenn Sie eine Kritik über meine Antwort, Sie können sehen, dass die iterkeys/itervalues Methoden spart Ihnen die Schaffung einer Reihe von Provisorien, auch.
    InformationsquelleAutor Hugh Bothwell
Schreibe einen Kommentar