python - 在python中获取列表的旋转

Question

我想要所有可能的单右移数组:

def gen(total):
    rot = []
    init = [0]*total
    init[0] = 1
    print init
    rot.append(init)
    for i in range(total-1):
        init[i] = init[i] - init[i+1]
        init[i+1] = init[i] + init[i+1]
        init[i] = init[i+1] - init[i]
        print init
        rot.append(init)
    return rot

rot = gen(8)

print rot

这打印

[1, 0, 0, 0, 0, 0, 0, 0]
[0, 1, 0, 0, 0, 0, 0, 0]
[0, 0, 1, 0, 0, 0, 0, 0]
[0, 0, 0, 1, 0, 0, 0, 0]
[0, 0, 0, 0, 1, 0, 0, 0]
[0, 0, 0, 0, 0, 1, 0, 0]
[0, 0, 0, 0, 0, 0, 1, 0]
[0, 0, 0, 0, 0, 0, 0, 1]

我希望这是列表列表。我初始化一个数组 rot 但附加到它会创建相同的行

[[0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 1], [0, 0, 0, 0, 0, 0, 0, 1]]

如何处理这个问题？

PS:我不想像numpy.eye(8)那样使用numpy

Answer 1

现有的答案很好地涵盖了现有代码的问题，所以这里有一个完全不同的方法:

>>> [[int(a==b) for b in range(8)] for a in range(8)]
[[1, 0, 0, 0, 0, 0, 0, 0], [0, 1, 0, 0, 0, 0, 0, 0], 
[0, 0, 1, 0, 0, 0, 0, 0], [0, 0, 0, 1, 0, 0, 0, 0], 
[0, 0, 0, 0, 1, 0, 0, 0], [0, 0, 0, 0, 0, 1, 0, 0], 
[0, 0, 0, 0, 0, 0, 1, 0], [0, 0, 0, 0, 0, 0, 0, 1]]

(为了易读性我在输出中添加了一些换行符)