首頁技術文章正文

用Python生成一個上三角矩陣

更新時間:2023-02-21 來源:黑馬程序員 瀏覽量:

IT培訓班

  可以使用Python來生成一個上三角矩陣,可以使用嵌套列表或NumPy數組實現(xiàn)。

  使用嵌套列表的方法如下:

n = 5  # 矩陣的維度
matrix = [[0] * n for i in range(n)]  # 創(chuàng)建一個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))  # 創(chuàng)建一個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.]]


分享到:
在線咨詢 我要報名
和我們在線交談!