Python Sequenzelemente Vergleich

Question

Ich habe diese beiden Funktionen:

def comparison(a, b): 
    return [-1, 0, 1].index(cmp(b, a)) 

def base_3(seq): 
    return [comparison(a, b) for a, b in itertools.combinations(seq, 2)] 

Die Funktion Vergleich liefert eine Vergleichszahl auf dieser Basis:

1. 0, wenn a> b
2. 1, wenn a = b
3. 2, wenn a < b

Die Funktion base_3 gibt den Vergleich zwischen allen kombinierten Elementen zurück.

Zum Beispiel:

x = [0, 1, 2] 
y = [1, 2, 0] 
z = [0, 1, 0] 

>>> base_3(x) 
[2, 2, 2] 

>>> base_3(y) 
[2, 0, 0] 

>>> base_3(z) 
[2, 1, 0] 

Ich brauche eine Funktion, die eine Sequenz aus einem gegebenen base_3 zurückkehren, wenn möglich:

>>> base_3_to_seq([2, 2, 2]) 
[0, 1, 2] 

>>> base_3_to_seq([2, 1, 0]) 
[0, 1, 0] 

>>> base_3_to_seq([0, 2, 1]) 
"Impossible" 

Wie kann ich diese Funktion base_3_to_seq schreiben?

Answer 1

Dies ist eine Brute-Force-Weg:

def base_3_to_seq(base3): 
    for seq in itertools.product(range(3), repeat=3): 
     if base_3(seq) == base3: 
      return seq 
    return "Impossible" 

Abhängig davon, welchen Weg Sie es sind nennen die meisten, könnten Sie den Code oben für lange Läufe beschleunigen, indem entweder base_3 oder base_3_to_seq memoizing.