Question

我只是想输出set的内容（因为我有使用这个被诅咒和不灵活的类型），但每次我去做或循环通过打印每个元素，我得到最后一个，而不是其余的。这是我的整个代码：

def factorial(n):
    r = 1
    for i in range(1, n + 1):
        r *= i
    return r

def nCr(n, r):
    return factorial(n) / (factorial(r) * factorial(n - r))

def combinations(s, tokens):
    maxcombos = int(nCr(len(s), tokens))
    for index in range(maxcombos):
        token_copy = tokens
        combo = []
        for val in range(1, len(s) + 1):
            if not token_copy:
                break
            thresh = nCr(len(s) - val, token_copy - 1)
            if index < thresh:
                combo.append(s.pop())
                token_copy -= 1
            else:
                index -= thresh
        yield tuple(combo)

def powerset(s):
    pset = set()
    for num_tokens in range(1, len(s)):
        for combo in combinations(s, num_tokens):
            pset.add(combo)
    pset.update(s)
    #pset.update(tuple())
    return pset

if __name__ == '__main__':
    s = set([1,2,23])
    p = powerset(s)
    msg = "Initial set: set(["
    for (i, item) in enumerate(s):
        msg += str(item) + ", "
    msg += "])"
    print(msg)
    #print(powerset())

我只想让每一个出现在([])之间，就像普通集一样。有没有办法简单地输出整个变量及其内容？谢谢！

Answer 1

您的powerset函数实际上会改变其输入。在combinations中，您使用s.pop()。这几乎肯定是错的，并且可以解释为什么当你试图打印它时它的内容已经改变了。

最简单的解决方法是使用powerset文档中的配方替换itertools函数： http://docs.python.org/2/library/itertools.html#recipes

from itertools import chain, combinations
def powerset(input_set):
    "powerset([1,2,3]) --> () (1,) (2,) (3,) (1,2) (1,3) (2,3) (1,2,3)"
    s = list(input_set)
    return chain.from_iterable(combinations(s, r) for r in range(len(s)+1))

Answer 2

将组合定义的顶部更改为以下内容：

def combinations(r, tokens):
    s = set(r)
    # everything else you had

这会创建一个新组，在组合函数中不会更改。它之前被改变的原因是因为组合中的局部变量是对函数外部的同一对象的引用。如果两个名称引用同一个对象，则改变一个会影响另一个。

在主要内容中，您需要以下内容：

for (i, item) in enumerate(s):
    msg += str(item) + ", "
msg = msg[:-2]
msg += "])"

Python中的输出集及其内容？

2 个答案: