在Numpy 2d数组上应用RLE

Question

我有一个像这样的numpy 2D数组：

np.array([[1,1,1,0], [1,0,0,1]])

如何有效地在这个2D阵列上应用RLE？我的数据集的形状是（4000,3000）

我能够在不使用numpy的情况下使用此逻辑对字符串执行操作。

    for i in new_bin_data:
        if i == '0':
            if prev != i:
                final_result.append(count)
                count = 0
                prev = '0'
            count += 1
        else:
            if prev != i:
                final_result.append(count)
                count = 0
            count += 1
            prev = '1'