numpy.diag 函数在不同版本的 NumPy 中可能存在一些兼容性问题
NumPy 1.7 及更早版本：在这些版本中，numpy.diag 函数只接受一个参数，即要提取对角线元素的矩阵。如果你需要从多维数组中提取对角线元素，请使用 numpy.diagonal 函数。
NumPy 1.8 及更高版本：在这些版本中，numpy.diag 函数已经得到了改进，现在可以接受两个参数。第一个参数是要提取对角线元素的矩阵，第二个参数是可选的，表示要提取的对角线的偏移量。默认值为 0，表示主对角线。正值表示上对角线，负值表示下对角线。
为了确保代码的兼容性，你可以根据 NumPy 的版本来选择合适的函数。例如：
import numpy as np
if np.__version__ < '1.8':
# 使用 numpy.diagonal 函数
diagonal_elements = np.diagonal(matrix)
else:
# 使用 numpy.diag 函数
diagonal_elements = np.diag(matrix)
或者，你可以直接使用 numpy.diag 函数，并在需要时提供第二个参数来指定对角线的偏移量。这样，你的代码将在 NumPy 1.7 及更高版本中都能正常工作。
