使用`组合'进行3种或更多种组合

时间:2013-06-03 19:50:01

标签: python combinations itertools

以下是我在Sage中编写的一些代码来计算某些李代数的维数,这些代数等于$ p ^ 2 $,对于某些$ p $。

def A_comb2rep(p):
    bound = p*p
    name_fund = []
    name_comb = []
    A = lambda i: WeylCharacterRing("A{0}".format(i))
    for i in range(bound):
        for k in range(1,bound+1):
            fw = A(i+1).fundamental_weights()
            if A(i+1)(k * fw[1]).degree() > bound:
                break
            else:
                for v in fw:
                    if A(i+1)(k * v).degree() == bound:
                        name_fund.append([])
                        name_fund[len(name_fund)-1].append('A'+str(i+1)+'('+str(k)+'*'+str(v)+')')
    for i in range(1,bound): # now onto combinations of two of the fws   #####
        fw = A(i+1).fundamental_weights()
        for k in fw:
            if A(i+1)(fw[1] + fw[2]).degree() > bound:
                break
            else:
                for j in fw:
                    rep = A(i+1)(j+k)
                    deg = rep.degree()
                    if deg == bound:
                        name_comb.append([])
                        name_comb[len(name_comb)-1].append('A'+str(i+1)+'['+str(j)+'+'+str(k)+']')
    return name_comb, name_fund

代码的后半部分是我考虑两个基本权重的组合。我现在想知道如何使用iterables模块中的combination函数对3个或更多基本权重的组合进行扩展。

更具体地说,我如何编码fw的3个元素的总和?我知道v = combinations(fw, 3)会将v所有$ {n \ choose 3} $三元组合放入fw,但(1,1,1,0,0,0)的元素是元组,如v。那么我如何对{{1}}中得到的每个三元组进行求和?如果这个问题不适合本网站,我深表歉意。

1 个答案:

答案 0 :(得分:1)

我不擅长数学,但如果问题是“我怎样才能将我得到的每个三元组合起来?”然后答案是:

sv = map(sum, v)

其中sv将包含来自v

的三元组总和的列表(实际上是一个迭代器)