找到具有一定数量的变量和度数的所有单项式

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

有没有更好的算法来做到这一点？

修改

1）假设＆＃39; xz＆＃39; ！=＆＃39; zx＆＃39; 2）假设＆＃39; xx＆＃39; =＆＃39; x ^ 2＆＃39;

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