测试数组的最后第n行是否全部为零

Question

我有这个数组：

A = np.array([[[ 1.8, -3.1, -3.5,  2.2],
[ 1.5, -6.6,  1.1,  1.1],
[ 8.9,  4.8, -1.2,  3.6],
[ 1.3, -7.4,  7.4,  1. ],
[ 6.3,  0. ,  0. ,  3. ],
[ 6.3,  0. , -6.3,  0. ],
[ 6.3, -6.3,  6.3,  3.3],
[ 0. ,  0. ,  0. ,  0. ],
[ 0. ,  0. ,  0. ,  0. ],
[ 0. ,  0. ,  0. ,  0. ]]])

所以我想检查这个数组的最后3行是否都为零：

counter = 0
if A[last-3-rows==0]:
counter += 1

Answer 1

您可以使用np.all检查具有以下索引的最后三行：

>>> np.all(A[:, -3:] == 0)
# or alternatively
# >>> (A[:, -3:] == 0).all()
True

如果你想把它作为一个整数：

>>> np.all(A[:, -3:] == 0).astype(int)
1