生成一定次数的所有多项式项

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')]