Question

我有一个看起来像的矩阵：

x=[[a,b,c,d,e,f],
   [g,h,i,j,k,l],
   [m,n,o,p,q,r]]

带a，b，c数字。然而，我只对左下半部分的数字感兴趣，并希望创建一个下三角矩阵，它按两个位置的步长，因此看起来像这样：

x2=[[a,b,0,0,0,0],
    [g,h,i,j,0,0],
    [m,n,o,p,q,r]]

我当然可以将x乘以：

x3=[[1,1,0,0,0,0],
    [1,1,1,1,0,0],
    [1,1,1,1,1,1]]

但有没有办法在不手动创建x3的情况下执行此操作？

是否有可能创建一个脚本，其中步骤一次大于2个零？

Answer 1

使用您在示例中提供的示例矩阵x3，您可以执行以下操作：

x3=[[1,1,1,1,1,1],
    [1,1,1,1,1,1],
    [1,1,1,1,1,1]]


for i in range(len(x3)):
    step = (i+1) * 2
    for j in range(step, len(x3[i])):
        x3[i][j] = 0

for i in x3:
    print(i)

输出：

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

或者如果你喜欢单行：

x3 = [[0 if j>(i+1)*2 else x3[i][j] for j in range(0, len(x3[i]))] for i in range(len(x3))]

如何使用两个位置的步长制作下三角矩阵

1 个答案: