如何将 python 数组裁剪为仅非零值

Question

我有一个 numpy 像素数据数组，比如

0 0 0 0 0 0 0
0 1 3 4 6 1 0
0 2 3 5 2 1 0
0 1 0 0 1 0 0
0 0 0 0 0 0 0

我想得到一个新数组，它排除任何带零的外部行/列，所以我最终只得到非零值，即

1 3 4 6 1
2 3 5 2 1

到目前为止，我所设法得到的是

1 3 4 6 1
2 3 5 2 1
1 0 0 1 0

使用 np.argwhere 查找“最小”和“最大”非零值，但这仍然包括具有零值和非零值的行/列。

Answer 1

欢迎来到 StackOverflow！

输入：

pixles_input = np.array([[0, 0, 0, 0, 0, 0, 0],
                         [0, 1, 3, 4, 6, 1, 0],
                         [0, 2, 3, 5, 2, 1, 0],
                         [0, 1, 0, 0, 1, 0, 0],
                         [0, 0, 0, 0, 0, 0, 0]])

说明

pixles_input = pixles_input[1:-1, 1:-1]

# Convert pixles_input to list of lists
pixles_input = pixles_input.tolist()

# remove sub list that contains a zero
pixles_input = [sub for sub in pixles_input if 0 not in sub]

# Convert pixles_input to numpy array
pixles_input = np.array(pixles_input)

输出：

[[1 3 4 6 1],
[2 3 5 2 1]]