请注意以下代码:
class permcom:
def __init__(self, INPUT_SET, IS_PERM, REPETITION):
self.end_set = []
self.input_set = INPUT_SET
self.is_perm = IS_PERM
self.repetition = REPETITION
def helpfunc(self, seen, depth, current):
if depth == 0:
self.end_set.append(seen)
else:
for i in range(0, len(self.input_set)):
if(self.repetition):
seen.append(self.input_set[i])
if(self.is_perm):
self.helpfunc(seen, depth - 1, 0)
else:
self.helpfunc(seen, depth - 1, i)
del seen[-1]
# return all permutations with repetition
def rapwr(INPUT_SET, subset_size):
instance = permcom(INPUT_SET, True, True)
A = []
instance.helpfunc(A, subset_size, 0)
return instance.end_set
A = [1,2,3]
B = rapwr(A, 2)
for i in range(0, len(B)):
print B[i]
输出如下:
[]
[]
[]
[]
[]
[]
[]
[]
[]
但是,预期的输出是:
[1, 1]
[1, 2]
[1, 3]
[2, 1]
[2, 2]
[2, 3]
[3, 1]
[3, 2]
[3, 3]
我花了太多时间看这段代码,不幸的是,我仍然无法弄清楚到底是什么错。必须有一些基本的东西,我不理解成员变量如何在Python中工作,但我仍然不太了解这里发生的事情以及为什么代码不起作用。有人可以解释一下吗?
答案 0 :(得分:2)
简短回答
您需要的是列表切片[:]
。改变陈述
if depth == 0:
self.end_set.append(seen)
到
if depth == 0:
self.end_set.append(seen[:])
给出预期答案
答案很长
在python解释器中尝试此示例代码
a = [1,2]
b = []
b.append(a)
a[0] = 3
print b
# output is [[3, 2]]
现在试试这段代码
a = [1,2]
b = []
b.append(a[:])
a[0] = 3
print b
# output is [[1, 2]]
为什么会这样?在第一种情况下,当您将a
添加到列表b
时,它不是附加的a
的值,而是参考/ tag 到[1,2]
值。您可以通过打印id(b[0])
和id(a)
来验证这一点。 两者都是相同的值。因此,当您修改a
列表中的任何值时,b
列表中的值也会更改。
您的代码中的情况也是如此。由于您正在执行del seen[-1]
,因此self.end_set
中的相应值也会被删除。您可以通过在self.end_set
块中打印depth == 0
的值来确认这一点。
要避免这种情况,请将一个列表的 clone 附加到另一个列表中。这是通过使用拼接语法[:]
完成的。这将从列表的开头到结尾创建列表的副本。您可以了解有关切片here的更多信息。
PS:尝试使用切片时打印两个列表中的id()
,值将不同
这是我得到的
a = [1,2]
b = []
b.append(a)
print id(b[0])
#output is 43337352L
print id(a)
#output is 43337352L
b = []
b.append(a[:])
print id(b[0])
#output is 43337608L
看一下这个python memory model diagram,以便更好地理解上面的
更新:一些建议
B
和self.input_set
都是列表,因此请使用惯用语for i in B
和for i in self.input_set
。rapwr
可以重命名为return_all_permutations_with repetition
。虽然名称很大,但是现在很容易理解它的作用而不用查看方法体。