如何获得 3D NumPy 数组的所有 2D 对角线？

在处理多维数组时，我们有时需要获取3D数组中所有2D对角线的值。这篇文章将介绍如何在NumPy中使用切片和数组操作来获得3D NumPy数组的所有2D对角线。

使用切片操作

我们可以使用切片操作来获得3D NumPy数组中所有2D对角线的值。对于一个形状为（M，N，K）的3D数组，我们可以通过对数组进行以下操作来获得所有2D对角线的值：

import numpy as np

# 创建一个形状为（3，3，3）的3D数组
arr = np.arange(27).reshape((3,3,3))

# 获得3D数组中所有2D对角线的值
diagonals = [arr[:, i, i] for i in range(arr.shape[1])]

此代码将创建一个形状为（3，3，3）的3D数组，并使用for循环遍历所有2D对角线。我们使用切片操作来选择第i列和第i行，并将它们添加到一个列表中。

数组操作

我们也可以使用数组操作来获得3D NumPy数组中所有2D对角线的值。我们可以使用NumPy的diag函数来获得2D数组的对角线，并使用它来选择3D数组中的所有对角线。对于一个形状为（M，N，K）的3D数组，我们可以通过对数组进行以下操作来获得所有2D对角线的值：

import numpy as np

# 创建一个形状为（3，3，3）的3D数组
arr = np.arange(27).reshape((3,3,3))

# 获得3D数组中所有2D对角线的值
diagonals = [np.diag(arr[:, :, i]) for i in range(arr.shape[2])]

此代码将创建一个形状为（3，3，3）的3D数组，使用for循环遍历所有2D对角线，并使用NumPy的diag函数来选择每个2D数组的对角线。我们使用切片操作来选择第i个2D数组，并将它们添加到一个列表中。

这就是获得3D NumPy数组中所有2D对角线的两种方法。我们可以使用切片操作或数组操作来实现这个任务。使用这两个技术之一，我们可以很容易地检索3D NumPy数组中的所有对角线值，并对它们执行操作。