主页 / 问题 / 1423320
Accepted
Gawain
Asked:2022-08-24 02:05:03 +0000 UTC2022-08-24 02:05:03 +0000 UTC

从numpy矩阵的上方和下方删除零行

  • 772

是否有一些非 hacky 方法可以从 numpy 矩阵的上方和下方删除零行？例子

输入：[[0, 0], [0, 0], [1, 2], [3, 4], [0, 0], [0,0]

输出：[[1, 2], [3, 4]]

python numpy

1 个回答

  1. Best Answer

    发生了这样的事情：

    import numpy as np

arr = np.array([[0, 0], [0, 0], [1, 2], [3, 4], [0, 0], [0,0]])
cond = np.nonzero(~(arr == 0).all(axis=1))[0]
arr = arr[cond[0]:cond[-1]+1]
print(arr)

    结论：

    [[1 2]
 [3 4]]

    更新：甚至像这样的东西也有效。我不太了解工作的逻辑np.nonzero，但显然仅此功能就足够了：

    cond = np.nonzero(arr)[0]
arr = arr[cond[0]:cond[-1]+1]
    • 1

相关问题