查找二进制数组中连续的个数答案

【问题标题】：Find number of consecutive ones in binary array查找二进制数组中连续的个数
【发布时间】：2015-05-01 17:47:30
【问题描述】：

我想在 MATLAB 的逻辑数组中找到所有系列的 1 和 0 的长度。这就是我所做的：

A = logical([0 0 0 1 1 1 1 0 1 1 0 0 0 0 0 0 1 1 1 1 1]);
%// Find series of ones:
csA = cumsum(A);
csOnes = csA(diff([A 0]) == -1);
seriesOnes = [csOnes(1) diff(csOnes)];
%// Find series of zeros (same way, using ~A)
csNegA = sumsum(~A);
csZeros = csNegA(diff([~A 0]) == -1);
seriesZeros = [csZeros(1) diff(csZeros)];

这可行，并给出seriesOnes = [4 2 5] 和seriesZeros = [3 1 6]。但是在我看来它是相当丑陋的。

我想知道是否有更好的方法来做到这一点。性能不是问题，因为它很便宜（A 不超过几千个元素）。我正在寻找代码清晰和优雅。

如果没有更好的办法，我就把它放在一个小辅助函数中，这样我就不用看了。

【问题讨论】：

标签： arrays matlab run-length-encoding

【解决方案1】：

这种基于strfind（与数字数组和字符串数组完美结合）的方法可能更容易理解 -

%// Find start and stop indices for ones and zeros with strfind by using
%// "opposite (0 for 1 and 1 for 0) sentients" 
start_ones = strfind([0 A],[0 1]) %// 0 is the sentient here and so on
start_zeros = strfind([1 A],[1 0])
stop_ones = strfind([A 0],[1 0])
stop_zeros = strfind([A 1],[0 1])

%// Get lengths of islands of ones and zeros using those start-stop indices 
length_ones = stop_ones - start_ones + 1
length_zeros = stop_zeros - start_zeros + 1

【讨论】：

【解决方案2】：

您可以为 run-length-encoding 使用现有代码，它会为您（丑陋的）工作，然后自己过滤掉您的向量。这样，您的辅助函数就相当通用，其功能从名称 runLengthEncode 中就可以看出。

重用来自this answer的代码：

function [lengths, values] = runLengthEncode(data)
startPos = find(diff([data(1)-1, data]));
lengths = diff([startPos, numel(data)+1]);
values = data(startPos);

然后您可以使用以下方法过滤掉您的向量：

A = logical([0 0 0 1 1 1 1 0 1 1 0 0 0 0 0 0 1 1 1 1 1]);
[lengths, values] = runLengthEncode(A);
seriesOnes = lengths(values==1);
seriesZeros = lengths(values==0);

【讨论】：

我最喜欢这个答案的普遍性。我知道我最初是在寻求清晰和优雅，其他答案也有这个，但我认为这个答案的额外普遍性值得接受。谢谢！
为了使上述RLE算法与行向量和列向量兼容，可以将函数的第一行更改为startPos = find(diff([data(1)-1; data(:)])).';。

【解决方案3】：

你可以试试这个：

A = logical([0 0 0 1 1 1 1 0 1 1 0 0 0 0 0 0 1 1 1 1 1]);
B = [~A(1) A ~A(end)];                %// Add edges at start/end
edges_indexes = find(diff(B));        %// find edges
lengths = diff(edges_indexes);        %// length between edges

%// Separate zeros and ones, to a cell array
s(1+A(1)) = {lengths(1:2:end)};
s(1+~A(1)) = {lengths(2:2:end)};

【讨论】：