產生一定程度上的所有多項式項

Question

有n項

ts = ['t1','t2','t3',...,'tn'] 

的名單有實現這一條款的所有可能的Q-長度組合的任務。

因此，

ts = ['t1','t2'] 
q = 4 

答案是

[['t1','t1','t1','t1'],['t1','t2','t2','t2'],['t1','t1','t2','t2'], 
['t1','t1','t1','t2'],['t2','t2','t2','t2']] 

這裏是我的解決方案

powers = np.array(list(itertools.product(*[range(q+1)]*len(ts)))) 
powers = powers[np.where(np.asarray(map(sum, powers))==q)] 
res = map(lambda ps_: flatten([ p*[t] for p,t in zip(ps_,ts) ]), powers) 

其中flatten是從列表的列表產生列表功能。

雖然有解決方案，產生所需的結果，我想知道是否有更簡單的方法來解決這個問題？

Answer 1

你需要的是與替換組合。最簡單的解決方案是使用恰當的名稱itertools.combinations_with_replacement：

>>> list(itertools.combinations_with_replacement(ts,q)) 
[('t1', 't1', 't1', 't1'), 
('t1', 't1', 't1', 't2'), 
('t1', 't1', 't2', 't2'), 
('t1', 't2', 't2', 't2'), 
('t2', 't2', 't2', 't2')]