测试numpy数组是否对称？

Question

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

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

我确定我错过了一个(duh)答案,但是这里的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()

Answer 2

如果您的数组包含浮点数(特别是如果它们是计算的结果),请使用 allclose

np.allclose(arr.transpose(1, 0, 2), arr)

如果您的某些值可能是NaN,请在测试之前将它们设置为标记值.

arr[np.isnan(arr)] = 0