ある程度の変数と次数を持つ単項式をすべて見つけてください

Question

[x、y、z]のようなリストと2のような数字があるとします。 この3変数と次数= 2のすべての単項式を見つけるアルゴリズムが必要です

私のアルゴリズム：

def mul(x:list, y:list) -> list: 
     return ["".join(elm) for elm in product(x, y)] 

def f(x:list, n:int) -> list: 
    r = x; 
    for i in range(n-1): 
     r = mul(r, x) 
    return r 

>>> f(['x','y','z'],2) 
['xx', 'xy', 'xz', 'yx', 'yy', 'yz', 'zx', 'zy', 'zz'] 

これを行うには、任意のより良いアルゴリズムがありますか？

EDIT：！

1） 'XZ' = 'ZX' 2と仮定）と仮定X-X '' = 'X^2'

Answer 1

私は、あなたが探しているが呼ばれると信じてproductモジュールitertoolsから

from itertools import product 

a=product(['x','y','z'], repeat=2) 
list(map(lambda x: ''.join(x), a)) 

出力：

['xx', 'xy', 'xz', 'yx', 'yy', 'yz', 'zx', 'zy', 'zz']