如何访问N*M*Mnumpy数组的下三角？

下面的代码展示了如何访问 NMM numpy 数组的下三角。这可以通过使用 numpy 的 tril 函数和布尔索引来实现。

import numpy as np

# 创建一个 3*3*3 的 numpy 数组
arr = np.array([[[1, 2, 3], [4, 5, 6], [7, 8, 9]],
                [[10, 11, 12], [13, 14, 15], [16, 17, 18]],
                [[19, 20, 21], [22, 23, 24], [25, 26, 27]]])

# 访问下三角
lower_triangle = np.tril(arr)

# 打印结果
print(lower_triangle)

这将输出以下结果：

array([[[ 1,  0,  0],
        [ 4,  5,  0],
        [ 7,  8,  9]],

       [[10,  0,  0],
        [13, 14,  0],
        [16, 17, 18]],

       [[19,  0,  0],
        [22, 23,  0],
        [25, 26, 27]]])

可以看到，在每个 3*3 的子数组中，只有下三角的部分包含数值，其他部分都为零。