python - 在numpy数组中找到正/负/正模式

Question

numpy 是否有一种简单的方法来确定数组是否具有如下所示的模式：（正数后跟负数后跟正数）？

例子：

myArray = np.array([5, 5, 3, 6, -2, -5, 4, 9])
myArray2 = np.array([5, 5, 3, 6, 8, 4, -5, -8])

if foundPatern(myArray):  # True
if foundPatern(myArray2): # False

Answer 1

有一个 numpy 内置函数可以根据符号将数组的每个元素转换为 1、0、-1。转换符号后，您可以轻松地使用itertools.groupby 完成此任务：

from itertools import groupby
import numpy as np


def foundPattern(lst):
    g = list(groupby(np.sign(lst)))
    return (len(g)==3 and g[0][0]==1.0)

它的工作原理如下：

1. 数组中的项目根据其符号（+ 或 -）进行分组
2. 如果恰好存在 3 个组并且第一个是一组阳性，则返回 True；在所有其他情况下，您返回False

请注意，您尚未指定如何处理边缘情况（例如，数组中的 0），因此我不会尝试以任何方式处理它们。但是请注意它们确实存在。另一个极端情况的例子可能是NaN。