有一个三维的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]
...
,看看是否所有边值都是零。
答案 0 :(得分:1)
:
与slice(None, None, None)
A[0, -1, :]
与
obj = (0, -1, slice(None, None, None))
A[obj]