更新时间:2023-02-21 来源:黑马程序员 浏览量:
可以使用Python来生成一个上三角矩阵,可以使用嵌套列表或NumPy数组实现。
使用嵌套列表的方法如下:
n = 5 # 矩阵的维度 matrix = [[0] * n for i in range(n)] # 创建一个n x n的全零矩阵 for i in range(n): for j in range(i, n): matrix[i][j] = 1 # 上三角部分赋值为1 print(matrix)
输出结果为:
[[1, 1, 1, 1, 1], [0, 1, 1, 1, 1], [0, 0, 1, 1, 1], [0, 0, 0, 1, 1], [0, 0, 0, 0, 1]]
使用NumPy数组的方法如下:
import numpy as np n = 5 # 矩阵的维度 matrix = np.zeros((n, n)) # 创建一个n x n的全零矩阵 for i in range(n): for j in range(i, n): matrix[i][j] = 1 # 上三角部分赋值为1 print(matrix)
输出结果为:
[[1. 1. 1. 1. 1.] [0. 1. 1. 1. 1.] [0. 0. 1. 1. 1.] [0. 0. 0. 1. 1.] [0. 0. 0. 0. 1.]]