列表列表而不是元组列表 - python

时间:2013-06-17 20:59:26

标签: python list tuples itertools

我正在开发一个程序,它将提供值列表的所有组合而不重复。只要[10,9,8]和[8,9,10]都不可能,结果如何排序并不重要。到目前为止,我有一段代码可以完成我想要的,但我希望结果是以列表格式而不是元组。它使用sage库中的unordered_tuples函数,但所有这些都返回一个列表列表,如下所示:

>from sage.all import unordered_tuples
>tuples = unordered_tuples([10,9,8],2) 
>print tuples
[[8, 8], [8, 9], [8, 10], [9, 9], [9, 10], [10, 10]] 

from itertools import product
def combos():
    dicti = {} 
    index = 0 
    for entry in [1,0,1,0]: 
        dicIndex = str('0')+str(index) 
        print dicIndex 
        if entry == 0: dicti[dicIndex] = [[0]] 
        else: dicti[str('0')+str(index)] = unordered_tuples([10,9,8],entry) 
        index += 1 
    lis = ['00','01','02','03'] 
    value = dicti[lis[0]] 
    print dicti 
    for index in lis[1:]: 
        value = product(value, dicti[index]) 
        if index == lis[-1]: 
            print list(value) 
            print 

输出:

[((([8], [0]), [8]), [0]), ((([8], [0]), [9]), [0]), ((([8], [0]), [10]), [0]), ((([9], [0]), [8]), [0]), ((([9], [0]), [9]), [0]), ((([9], [0]), [10]), [0]), ((([10], [0]), [8]), [0]), ((([10], [0]), [9]), [0]), ((([10], [0]), [10]), [0])] 

想要:

[[[8], [0], [8], [0]], [[8], [0], [9], [0]], [[8], [0], [10], [0]], [[9], [0], [8], [0]], [[9], [0], [9], [0]], [[9], [0], [10], [0]], [[10], [0], [8], [0]], [[10], [0], [9], [0]], [[10], [0], [10], [0]]]

2 个答案:

答案 0 :(得分:1)

output = [[[tuple[0]], [0], [tuple[1]], [0]] for tuple in unordered_tuples([10,9,8],2)]

答案 1 :(得分:0)

从PostgreSQL数据库中提取数据时遇到了类似的问题。这是我使用的方法。

def tupleToList(t):
    l = []
    for i in xrange(len(t)):
        l.append(list(t[i]))
        pass
    return l
    pass

这适用于2D列表。我正在努力将其推荐给你的5D列表。