python - 如何在重复数字的python中创建矩阵？

Question

我想:
创建一个从 0 到 4 的向量列表，即 [0, 1, 2, 3, 4] 并从那个
创建一个包含从 0 到 4 的“分层列表”的矩阵，重复 3 次，每个维度一次。矩阵有 4^3 = 64 行，例如

T = [0 0 0    
       0 0 1  
       0 0 2  
       0 0 3  
       0 0 4  
       0 1 0  
       0 1 1  
       0 1 2  
       0 1 3  
       0 1 4  
       0 2 0  
       ...  
       1 0 0  
       ...  
       1 1 0  
       ....  
       4 4 4]    

这是我目前所拥有的:

n=5;
ind=list(range(0,n))
print(ind)

我刚刚开始使用 Python，所以非常感谢任何帮助!

Answer 1

python itertools module product() 函数可以做到这一点:

for code in itertools.product( range(5), repeat=3 ):
    print(code)

给出结果:

(0, 0, 0)
(0, 0, 1)
(0, 0, 2)
(0, 0, 3)
...
(4, 4, 2)
(4, 4, 3)
(4, 4, 4)

所以把它变成一个矩阵:

import itertools
matrix = []
for code in itertools.product( range(5), repeat=3 ):
     matrix.append( list(code) )