Как найти последовательные положительные, отрицательные и нули в массиве numpy?

Question

Я использую следующую функцию, чтобы найти последовательные отрицательные и положительные числа, теперь я также хочу добавить условие, которое также получает последовательные нули.Как я могу это сделать?

def consecutive_counts(arr):
    '''
    Returns number of consecutive negative and positive numbers
    arr = np.array
    negative = consecutive_counts()[0]
    positive = consecutive_counts()[1]
    '''
    pos = arr > 0
    # is used to Compute indices that are non-zero in the flattened version of arr
    idx = np.flatnonzero(pos[1:] != pos[:-1])
    count = np.concatenate(([idx[0]+1], idx[1:] - idx[:-1], [arr.size-1-idx[-1]]))
    negative = count[1::2], count[::2]
    positive = count[::2], count[1::2]
    if arr[0] < 0:
        return negative
    else:
        return positive

это серия панд:

In [221]: n.temp.p['50000']
Out[221]: 
name
0         0.00
1       -92.87
2       -24.01
3       -92.87
4       -92.87
5       -92.87
...       ...

, которую я использую следующим образом:

arr = n.temp.p['50000'].values #Will be a numpy array as the input

ожидаемый результат:

In [225]: consecutive_counts(a)
Out[225]: (array([30, 29, 11, ...,  2,  1,  3]), array([19,  1,  1, ...,  1,  1,  2]))

Спасибо:)

Answer 1

Поскольку вы пометили pandas, вот один из подходов:

# random data
np.random.seed(1)
a = np.random.choice(range(-2,3), 1000)

# np.sign: + = 1, 0 = 0, - = -1
b = pd.Series(np.sign(a))

# b.head()
# 0    1
# 1    1
# 2   -1
# 3   -1
# 4    1
# dtype: int32

# sign blocks
blks = b.diff().ne(0).cumsum()

# blks.head()
# 0    1
# 1    1
# 2    2
# 3    2
# 4    3
# dtype: int32

# number of blocks:
blks.iloc[-1]
# 654

# block counts:
blks.value_counts()

# 1      2
# 2      2
# 3      1
# 4      3
# 5      2
# ...

Answer 2

Вот примерный подход:

# create example
arr = np.random.randint(-2,3,(10))

# split into negative, zero, positive
*nzp, = map(np.flatnonzero,(arr<0,arr==0,arr>0))
# find block boundaries
*bb, = (np.flatnonzero(np.diff(x,prepend=-2,append=-2)-1) for x in nzp)
# compute block sizes
*bs, = map(np.diff,bb)

# show all
for data in (arr,nzp,bb,bs): print(data)
# [-1  1 -1  1  0  0  2 -1 -2  1]
# [array([0, 2, 7, 8]), array([4, 5]), array([1, 3, 6, 9])]
# [array([0, 1, 2, 4]), array([0, 2]), array([0, 1, 2, 3, 4])]
# [array([1, 1, 2]), array([2]), array([1, 1, 1, 1])]