获取长方体的所有边(3D NumPy数组)

时间:2018-01-30 21:20:07

标签: python arrays numpy indexing

有一个三维的numpy数组A我想得到所有的(想象这个数组是一个长方体)。

好吧,A[0, 0, :]会给我一个优势,A[0, -1, :]第二个和A[:, -1, -1]又一个......所以我所要做的就是获得{{{{1}的所有排列1}},0-1并将它们用作索引。零和一个很容易,但我怎么用冒号呢?

我可以解决它很长的路,但它很难看,我打赌这有一些整洁的numpy解决方案。类似的东西:

:

最后我要做的是在所有边的集合上for indices in permutations([0, -1, ':']): edge = A[indices] ... ,看看是否所有边值都是零。

1 个答案:

答案 0 :(得分:1)

:slice(None, None, None)

相同

A[0, -1, :]

相同
obj = (0, -1, slice(None, None, None))
A[obj]