使用迭代器对数组进行切片

Question

我正在查看这个 python 代码，我需要一些解释：

arr = [0, 0, 0, 0, 1, 2, 3, 4,5]

arr = arr[next((i for i, x in enumerate(arr) if x != 0), len(arr)):]

此代码将从数组中删除前导零，我试图了解它是如何工作的。我知道我们创建了一个迭代器，它将迭代arr除 0 值之外的所有元素，然后只迭代直到数组的长度（不包括）。

但是这些索引是如何由 next 返回的，组合成一个数组呢？

Answer 1

让我们一步一步看代码。你想切掉最初的零。如果您知道第一个非零元素的索引n，则表达式看起来像arr = arr[n:]这基本上就是我们在这里所拥有的，使用n = next((i for i, x in enumerate(arr) if x != 0), len(arr)).通常，双参数形式的next将返回第二个参数作为标记，而不是StopIteration在迭代器用完时引发 a。这就是它的len(arr)用途。如果所有元素都为零，则表达式变为arr = arr[len(arr):]  # obviously empty如果有一个非零元素，next调用将找到它的索引（启用enumerate），并返回它。