python – 无论顺序如何,都生成重复列表

我想生成将列表中的索引与“插槽”相关联的组合.例如,(0,1)表示0和1属于同一时隙,而2属于另一个时隙. (0,1,1)表示1,2,3属于同一时隙,而0本身属于同一时隙.在此示例中,0和1只是识别这些插槽的方法,但不包含我的使用信息.

因此,对于我的目的,0)与(1,1)绝对相同,并且(0,1)等同于(1,0).

经典的笛卡尔积产生了许多我想要摆脱的重复.

这是我通过itertools.product获得的:

>>> LEN,SIZE = (3,1)
>>> list(itertools.product(range(SIZE+1),repeat=LEN))
>>>
[(0,0),1),(1,1)]

这就是我想得到的:

>>> [(0,1)]

使用小型列表很容易,但我不太明白如何使用更大的集合来完成此操作.你有什么建议吗?

如果不清楚,请告诉我,以便我澄清我的问题.谢谢!

编辑:基于Sneftel的答案,这个功能似乎有效,但我不知道它是否真的产生了所有结果:

def test():
    for p in product(range(2),repeat=3):
        j=-1
        good = True
        for k in p:
            if k> j and (k-j) > 1:
                good = False
            elif k >j:
                j = k
        if good:
            yield p
最佳答案
我首先提出以下意见:

>每个组合的第一个元素必须为0.
>第二个元素必须是0或1.
>第三个元素必须是0,1或2,但如果第二个元素为1则只能为2.

这些观察结果表明以下算法:

def assignments(n,m,used=0):
    """Generate assignments of `n` items to `m` indistinguishable
    buckets,where `used` buckets have been used so far.

        >>> list(assignments(3,1))
        [(0,0)]
        >>> list(assignments(3,2))
        [(0,1)]
        >>> list(assignments(3,3))
        [(0,2)]

    """
    if n == 0:
        yield ()
        return
    aa = list(assignments(n - 1,used))
    for first in range(used):
        for a in aa:
            yield (first,) + a
    if used < m:
        for a in assignments(n - 1,used + 1):
            yield (used,) + a

这将在几秒钟内处理您的用例(12项,5个桶):

>>> from timeit import timeit
>>> timeit(lambda:list(assignments(12,5)),number=1)
4.513746023178101
>>> sum(1 for _ in assignments(12,5))
2079475

这比你在答案结束时给出的函数(调用产品然后删除无效赋值的函数)快得多,如果它被修改为处理(12,5)用例:

>>> timeit(lambda:list(test(12,number=1)
540.693009853363

相关文章

本文从多个角度分析了vi编辑器保存退出命令。我们介绍了保存...
Python中的回车和换行是计算机中文本处理中的两个重要概念,...
SQL Server启动不了错误1067是一种比较常见的故障,主要原因...
信息模块是一种可重复使用的、可编程的、可扩展的、可维护的...
本文从电脑配置、PyCharm版本、Java版本、配置文件以及程序冲...
本文主要从多个角度分析了安装SQL Server 2012时可能出现的错...