更新時(shí)間:2023-02-21 來源:黑馬程序員 瀏覽量:
可以使用Python來生成一個(gè)上三角矩陣,可以使用嵌套列表或NumPy數(shù)組實(shí)現(xiàn)。
使用嵌套列表的方法如下:
n = 5 # 矩陣的維度 matrix = [[0] * n for i in range(n)] # 創(chuàng)建一個(gè)n x n的全零矩陣 for i in range(n): for j in range(i, n): matrix[i][j] = 1 # 上三角部分賦值為1 print(matrix)
輸出結(jié)果為:
[[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數(shù)組的方法如下:
import numpy as np n = 5 # 矩陣的維度 matrix = np.zeros((n, n)) # 創(chuàng)建一個(gè)n x n的全零矩陣 for i in range(n): for j in range(i, n): matrix[i][j] = 1 # 上三角部分賦值為1 print(matrix)
輸出結(jié)果為:
[[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.]]