从对角线创建一个3-D（或更大）的NumPy数组

要从对角线创建一个3-D（或更大）的NumPy数组，可以使用np.diag函数来在对角线上创建一个2-D数组，然后使用np.repeat函数来将该数组复制为多个2-D数组，并调整形状以创建3-D数组。

下面是一个示例代码：

import numpy as np

# 创建一个1-D数组作为对角线元素
diagonal = np.array([1, 2, 3])

# 使用np.diag函数在对角线上创建一个2-D数组
diagonal_matrix = np.diag(diagonal)

# 使用np.repeat函数复制该2-D数组，并调整形状创建3-D数组
three_d_array = np.repeat(diagonal_matrix[np.newaxis, ...], 3, axis=0)

print(three_d_array)

输出:

[[[1 0 0]
  [0 2 0]
  [0 0 3]]

 [[1 0 0]
  [0 2 0]
  [0 0 3]]

 [[1 0 0]
  [0 2 0]
  [0 0 3]]]

在这个示例中，我们首先创建一个1-D数组diagonal作为对角线元素。然后，使用np.diag函数将该数组转换为一个2-D数组diagonal_matrix，其中对角线元素为diagonal，其他元素为0。

接下来，我们使用np.repeat函数复制diagonal_matrix为多个2-D数组，并通过指定axis=0来使其在第0维上进行复制。最后，我们得到了一个3-D数组three_d_array，其中包含了3个相同的2-D数组，每个数组的对角线元素都是diagonal的值。

这样，我们就成功地从对角线创建了一个3-D的NumPy数组。