完成代码,输入是一个整数N,要求返回一个N行N列的矩阵,对角线元素值为N,其余位置值全为N-1。数组的数据类型为numpy.ndarray。
时间: 2024-02-25 17:55:29 浏览: 54
numpy.ndarray 交换多维数组(矩阵)的行/列方法
5星 · 资源好评率100%
以下是代码实现:
```python
import numpy as np
def generate_matrix(N):
matrix = np.ones((N, N)) * (N - 1)
np.fill_diagonal(matrix, N)
return matrix
```
可以通过以下方式调用该函数并打印结果:
```python
N = 5
matrix = generate_matrix(N)
print(matrix)
```
输出结果为:
```
[[5. 4. 4. 4. 4.]
[4. 5. 4. 4. 4.]
[4. 4. 5. 4. 4.]
[4. 4. 4. 5. 4.]
[4. 4. 4. 4. 5.]]
```
阅读全文