python - 在不使用 for 循环的情况下删除 numpy 数组的前导零

Question

如何在不使用 for 循环的情况下仅从 numpy 数组中删除前导零？

import numpy as np

x = np.array([0,0,1,1,1,1,0,1,0,0])

# Desired output
array([1, 1, 1, 1, 0, 1, 0, 0])

我写了下面的代码

x[min(min(np.where(x>=1))):] 

我想知道是否有更有效的解决方案。

Answer 1

您可以使用 np.trim_zeros(x, 'f')。

'f' 表示从前面修剪零。选项“b”将从后面修剪零。默认选项“fb”从两侧修剪它们。

x = np.array([0,0,1,1,1,1,0,1,0,0])
# [0 0 1 1 1 1 0 1 0 0]
np.trim_zeros(x, 'f')
# [1 1 1 1 0 1 0 0]