numpy-创建计数上部矩阵

Question

这就是问题：您需要一种算法，要求用户提供方阵的行数，然后用计数整数填充矩阵的上三角形。您的矩阵应该是 2D Numpy 数组。

注意：

由于矩阵是正方形大小，行数和列数相同，因此您可以首先使用算法应使用 for 循环的适当 numpy 方法将矩阵初始化为所有零条目。

在循环内部，一条消息应显示当前行和列索引以及要存储在该位置的值。

循环完成后，应显示整个矩阵 您的算法将提示用户输入矩阵大小并适应输入的大小

提示：如果您难以完成此任务，请将其分解为更简单的类似任务。

这些可能是：

• 用值 1 填充整个矩阵（在循环中）
• 打印一个方阵，其中整个矩阵用递增的整数填充

完成这些任务后，您可能会对如何解决该任务有更好的想法在这个问题中。

这是我的代码：

#input size of matrix from the user
M = int(input('Enter Matrix size: '))
N = M
#The dimension of the matrix in nXm 
dimension = (M, N)


#form a matrix of size nXm
matrix = np.zeros(dimension, dtype=int)

#use the num for filling the values in the matrix
num = 1
#for every row
for i in range (M):
    #the elements before the diagonal are 0 
    for j in range (i+1):
           
        #print ("The value at row  " + str(i) + " and column " + str(j) + " is 0", end = '\n')
    #the elements after the diagonal are num
        for j in range (i,N):
            #print ("The value at row " + str(i) + " and column " + str(j) + " is " + str(num), end = '\n')
            matrix[i][j] = num
        #increase num by 1
        num = num + 1
#print the upper triangular matrix
print (matrix)

它打印的矩阵是： [1, 1, 1 0, 3, 3 0, 0, 6]

虽然我想打印的矩阵是： [1,2,3 0, 4, 5 0, 0, 6]

Answer 1

我认为您想生成一个上三角方形矩阵。

import numpy as np


#input size of matrix from the user
n = int(input('Enter Matrix size: '))


#form a matrix of size nXn
matrix = np.zeros((n,n), dtype=int)

#use the num for filling the values in the matrix
num = 1
for i in range(n):
    #the elements before the diagonal are 0 
    for j in range(i):
      print (f"The value at row  {i} and column {j} is 0")

    for j in range (i, n):
      print (f"The value at row  {i} and column {j} is {num}")
      matrix[i][j] = num
      num = num + 1

print (f"matrix=\n{matrix}")

n = 4的输出

Enter Matrix size: 4
The value at row  0 and column 0 is 1
The value at row  0 and column 1 is 2
The value at row  0 and column 2 is 3
The value at row  0 and column 3 is 4
The value at row  1 and column 0 is 0
The value at row  1 and column 1 is 5
The value at row  1 and column 2 is 6
The value at row  1 and column 3 is 7
The value at row  2 and column 0 is 0
The value at row  2 and column 1 is 0
The value at row  2 and column 2 is 8
The value at row  2 and column 3 is 9
The value at row  3 and column 0 is 0
The value at row  3 and column 1 is 0
The value at row  3 and column 2 is 0
The value at row  3 and column 3 is 10
matrix=
[[ 1  2  3  4]
 [ 0  5  6  7]
 [ 0  0  8  9]
 [ 0  0  0 10]]