python - 有没有办法获得 3d 矩阵的 2d 切片(嵌套列表)

Question

我有一些数据存储在 3x3x3 嵌套列表中，我想从该列表中获取 2d 层。

例如，我想在以下列表中执行类似 a[:, :,2]、a[0,:,:] 或 a[:,1,:] 的操作:

a=[
[
    [(0, 0, 0), (0, 0, 1), (0, 0, 2)],
    [(0, 1, 0), (0, 1, 1), (0, 1, 2)],
    [(0, 2, 0), (0, 2, 1), (0, 2, 2)],
],
[
    [(1, 0, 0), (1, 0, 1), (1, 0, 2)],
    [(1, 1, 0), (1, 1, 1), (1, 1, 2)],
    [(1, 2, 0), (1, 2, 1), (1, 2, 2)],
],
[
    [(2, 0, 0), (2, 0, 1), (2, 0, 2)],
    [(2, 1, 0), (2, 1, 1), (2, 1, 2)],
    [(2, 2, 0), (2, 2, 1), (2, 2, 2)],
],]

获取:

[
[(0, 1, 0), (0, 1, 1), (0, 1, 2)],
[(1, 1, 0), (1, 1, 1), (1, 1, 2)],
[(2, 1, 0), (2, 1, 1), (2, 1, 2)],]

我尝试过，但是执行 a[:][1][:] 对列表没有任何作用，所以我只得到一个如下所示的 [1]:

[
[(1, 0, 0), (1, 0, 1), (1, 0, 2)],
[(1, 1, 0), (1, 1, 1), (1, 1, 2)],
[(1, 2, 0), (1, 2, 1), (1, 2, 2)],]

我认为我不能使用 numpy，因为我使用的是 jython，它不支持用 c 实现的库。

你能帮我找到一种很好的方法(或者找到一个像 numpy 这样适用于 jython 的库)吗？

Answer 1

这将收集每个子矩阵的中间行:

[x[1] for x in a]