美文网首页
用Python生成一个上三角矩阵

用Python生成一个上三角矩阵

作者: 鹤子青云上 | 来源:发表于2023-06-28 17:42 被阅读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.]]

    相关文章

      网友评论

          本文标题:用Python生成一个上三角矩阵

          本文链接:https://www.haomeiwen.com/subject/yygoydtx.html