如何将NumPy方阵永久转换为下三角矩阵？

我定义了一个NumPy矩阵a，编写的算法只能打印出下三角矩阵的形式，但直接打印原矩阵a时，它还是初始状态。我希望把矩阵a永久修改为下三角矩阵，而不是只打印这个形式，请问该怎么做？

原代码：

import numpy as np 
s = 5

# s是矩阵的行数和列数
a = np.array([[None]*s]*s)

print(a)
# 输出一个5行5列的矩阵


# 计算矩阵的行数和列数  
rows = len(a);  
cols = len(a[0]);  
   
if(rows != cols):  
    print("矩阵必须是方阵");  
else:  
    # 执行操作将矩阵转换为下三角矩阵（仅打印）
    print("下三角矩阵: ");  
    for i in range(0, rows):  
        for j in range(0, cols):  
            if(j > i):  
                print("0"),  
            else:  
                print(a[i][j]),  
      
        print(" ");  

# 到这里打印是正常的，能得到下三角矩阵的输出

print(a)
    
# 我原本希望这里打印的`a`已经是下三角矩阵，但结果和第一次打印的初始矩阵一样

你的代码只是在打印时临时输出0，并没有真正修改矩阵a的元素。要永久修改矩阵，需要在循环中直接对a的元素赋值，而不是只打印。

修改后的代码如下：

import numpy as np 
s = 5

# 更安全的矩阵创建方式，避免行引用问题
a = np.full((s, s), None)

print("初始矩阵:")
print(a)

rows = len(a)
cols = len(a[0])
   
if rows != cols:  
    print("矩阵必须是方阵")
else:  
    # 直接修改矩阵元素，将上三角部分设为0
    for i in range(rows):  
        for j in range(cols):  
            if j > i:  
                a[i][j] = 0  # 直接对矩阵元素赋值，而非仅打印
      
    print("\n修改后的下三角矩阵:")
    print(a)

# 再次打印验证
print("\n再次打印矩阵a:")
print(a)

关键说明 #

• 原代码中print("0")只是输出字符0，完全没有改变矩阵a的内容，所以原矩阵保持初始状态。
• 修改后的代码通过a[i][j] = 0直接对矩阵的上三角区域（j > i的位置）赋值为0，从根本上修改了矩阵a的存储内容。
• 额外优化：你原本用np.array([[None]*s]*s)创建矩阵的方式存在隐患——所有行都会指向同一个列表，修改某一行元素可能会影响其他行。改用np.full((s, s), None)可以避免这种引用问题。

内容的提问来源于stack exchange，提问作者sakabukayo