如何计算numpy数组中沿给定轴的连续Trues数?

使用布尔运算来识别 True 后面没有 真的 (使用切片和 pad ),那么 sum 这个 真的 每列:

out = ((a != np.pad(a[1:], ((0,1), (0,0)), constant_values=False))
       & a).sum(axis=0)

或者:

out = (a & ~np.pad(a[1:], ((0,1), (0,0)), constant_values=False)).sum(axis=0)

输出:

array([3, 2, 2])

可以通过沿第一个轴将元素与其后续元素进行比较,并将从False到True的转换添加到第一行的True值来完成此操作:

A[0,:] + (A[:-1,:]<A[1:,:]).sum(axis=0)

def count_consecutive_trues(arr, axis):
        consecutive_counts = []
        for row in arr:
                count = 0
                max_count = 0
                for value in row:
                        if value:
                                count += 1
                                max_count = max(max_count, count)
                        else:
                                count = 0
                consecutive_counts.append(max_count)
        return consecutive_counts