python - 在 nan 之后提取 numpy 数组中的第一个匹配项

Question

我有以下数组:

[1,1,1,1,1,1,nan,nan,nan,1,1,1,2,2,2,3,3]

我想在 nan 之后提取此数组中第一次出现的 1。我试过这个:

numpy.argmax(arr > numpy.nan)

Answer 1

np.where(np.isnan(foo))[0][-1] + 1

在 np.where 之后，0 返回包含 NaN 的元素的索引。然后 -1 给你最后一个 NaN 索引。然后将其加一以找到最后一个 NaN 之后的元素的索引。

在您的示例数组中，它生成的索引为 9

然后您可以再次使用 np.where 从索引 9 开始查找第一个 1。

总而言之:

afternan = np.where(np.isnan(foo))[0][-1] + 1np.where(foo[afternan:] == 1)[0][0]

请注意，您的示例似乎是一个列表。我假设您将其转换为 numpy 数组。