完成代码,输入为包含2个整数数字n,m的元组,返回一个列表格式的n*n的u矩阵,其对角线元素为m,其他位置为0 示例:(3,4) 输出·[[5,0,0],[0,5,0][0,0,5]] 限制:0<=x<=1000 0<=y<=1000
时间: 2024-03-22 10:41:17 浏览: 51
Python列表list解析操作示例【整数操作、字符操作、矩阵操作】
以下是Python代码实现:
```python
def create_matrix(n, m):
matrix = [[0 for i in range(n)] for j in range(n)]
for i in range(n):
matrix[i][i] = m
return matrix
n, m = (3, 4)
result = create_matrix(n, m)
print(result)
```
输出:
```
[[4, 0, 0], [0, 4, 0], [0, 0, 4]]
```
代码中,我们先创建一个 n*n 的矩阵 matrix,初始化所有元素为0。然后,我们遍历矩阵的对角线元素,将其赋为 m。最后,返回生成的矩阵 matrix。
阅读全文