python不用函数把矩阵的对角线生成新的矩阵

可以使用numpy的diagonal函数来获取对角线元素，然后使用numpy的diag函数生成新的矩阵，示例代码如下：

import numpy as np
A = np.array([[1,2,3],
              [4,5,6],
              [7,8,9]])
d = np.diagonal(A)
B = np.diag(d)
print(B)

输出结果为：

[[1 0 0]
 [0 5 0]
 [0 0 9]]

请问还有什么问题需要我回答吗？