Python - Match auf Array-Rückgabewert

Question

Ich möchte eine funktionale wie Musterübereinstimmung tun, um die ersten beiden Elemente zu erhalten, und dann den Rest eines Array-Rückgabewerts.

zum Beispiel angenommen, dass perms (x) eine Liste von Werten zurückgibt, und ich möchte, dies zu tun:

seq=perms(x) 
a = seq[0] 
b = seq[1] 
rest = seq[2:] 

Natürlich kann ich verkürzen:

[a,b] = seq[0:2] 
rest = seq[2:] 

Kann ich etwas Notation, um dies zu tun?

[a,b,more] = perms(x) 

oder konzeptuell:

[a,b,more..] = perms(x) 

PROLOG & funktionale Sprachen zu tun, so schön wie diese Zersetzung Liste!

Answer 1

Sie können es in Python 3 wie folgt tun:

(a, b, *rest) = seq 

Siehe extended iterable unpacking PEP für weitere Details.

Answer 2

Für Python 2, ich kenne Sie es mit einer Funktion tun können:

>>> def getValues(a, b, *more): 
    return a, b, more 

>>> seq = [1,2,3,4,5] 
>>> a, b, more = getValues(*seq) 
>>> a 
1 
>>> b 
2 
>>> more 
(3, 4, 5) 

aber nicht sicher, ob es eine Möglichkeit, es zu tun, wie Ayman's Python 3 suggestion

Answer 3

In Python 2, ist Ihre Frage sehr nahe auf eine Antwort bereits:

a, b, more = (seq[0], seq[1], seq[2:]) 

oder:

(a, b), more = (seq[0:2], seq[2:]) 

Answer 4

Sehr schön, danke.

Die Vorschläge, wo man das Array auf der Kampfseite seziert, funktionieren für mich nicht so gut, da ich eigentlich Muster auf die Renditen eines Generatorausdrucks anpassen wollte.

für (a, b, mehr) in perms (Seq): ...

Ich mag die P3-Lösung, sondern müssen warten, für Komodo, sie zu unterstützen!