python - 测试 numpy 数组是否对称？

Question

是否有更好的 pythonic 方法来检查 ndarray 在特定维度上是否对角对称？即对于所有 x

(arr[:,:,x].T==arr[:,:,x]).all()

我确定我错过了一个(呃)答案，但现在是 2 点 15 分...:)

编辑:澄清一下，我正在寻找一种更“优雅”的方式:

for x in range(xmax):
    assert (arr[:,:,x].T==arr[:,:,x]).all()

Answer 1

如果我没理解错的话，你想做检查

all((arr[:,:,x].T==arr[:,:,x]).all() for x in range(arr.shape[2]))

没有 Python 循环。方法如下:

(arr.transpose(1, 0, 2) == arr).all()