用Python生成一个上三角矩阵

2023-06-28  本文已影响0人  鹤子青云上

可以使用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.]]

上一篇下一篇

猜你喜欢

热点阅读