2013-05-09 22 views
6

Data una mappa sostitutiva come {search: replace, search: replace, ...} e una stringa, come generare un elenco di tutte le sostituzioni possibili di tale stringa (prima sottostringa sostituita, seconda sottostringa sostituita, entrambe sostituite ecc.). Esempio:Genera tutte le sostituzioni possibili

map = { 
    'bee': 'BETA', 
    'zee': 'ZETA', 
    'dee': 'DELTA' 
} 

source_string = 'bee foo zee bar bee' 

desired result = 
[ 
    'bee foo zee bar bee', 
    'BETA foo zee bar bee', 
    'bee foo ZETA bar bee', 
    'BETA foo ZETA bar bee', 
    'bee foo zee bar BETA', 
    'BETA foo zee bar BETA', 
    'bee foo ZETA bar BETA', 
    'BETA foo ZETA bar BETA' 
] 

L'ordine non è importante.

risposta

3

'bee foo zee bar bee' =>['bee', 'foo', 'zee', 'bar', 'bee']:

from itertools import product 

repl = { 
    'bee': 'BETA', 
    'zee': 'ZETA', 
    'dee': 'DELTA' 
} 
source_string = 'bee foo zee bar bee' 
p = product(*((x, repl[x]) if x in repl else (x,) for x in source_string.split())) 
for x in p: 
    print(x) 

uscita:

('bee', 'foo', 'zee', 'bar', 'bee') 
('bee', 'foo', 'zee', 'bar', 'BETA') 
('bee', 'foo', 'ZETA', 'bar', 'bee') 
('bee', 'foo', 'ZETA', 'bar', 'BETA') 
('BETA', 'foo', 'zee', 'bar', 'bee') 
('BETA', 'foo', 'zee', 'bar', 'BETA') 
('BETA', 'foo', 'ZETA', 'bar', 'bee') 
('BETA', 'foo', 'ZETA', 'bar', 'BETA') 
+0

Grazie, questo è quello che stavo cercando. – georg

3

Itertools.product potrebbe aiutare qui. Nel tuo esempio hai una scelta binaria per tre parole nella tua stringa. Così

itertools.product((0, 1), repeat=3) 

vi darà i vostri 8 possibili sostituzioni per ape e zee, dove 0 mezzi non sostituiscono e 1 mezzi sostituirli con BETA e ZETA rispettivamente.

Quanto segue fa quello che vuoi.

#!python3 

import itertools 

map = { 
    'bee': 'BETA', 
    'zee': 'ZETA', 
    'dee': 'DELTA' 
} 

source_string = 'bee foo zee bar bee' 

products = [] 
for word in source_string.split(): 
    if word in map: 
     products.append((word, map[word])) 
    else: 
     products.append((word,)) 

for words in itertools.product(*products): 
    print(' '.join(words)) 
+0

Grande, grazie! .. – georg

Problemi correlati