算法优化：python中打印任意矩阵

n = 6
lines = 2
width = n//lines
lst1 = []
lst2 = list(range(1,n+1))

for i in range(0,n,width):
    lst1.append(lst2[i:i+width])
else:
    for i in range(lines): 
        print(lst1[i])
print('---------------------------')

width = 2
lines = n//width
lst3 = []

# for _ in range(lines):
#     lst3.append([1]*width)
lst3 = [[1 for _ in range(width)] for _ in range(lines)]    #使用列表解析式一次性开辟空间比append效率要高

for x in range(width):
    for j in range(lines):
        lst3[j][x] = lst1[x][j]
else:
    for i in range(lines):
        print(lst3[i])

运行结果：

[1, 2, 3]
[4, 5, 6]
---------------------------
[1, 4]
[2, 5]
[3, 6]