2

在 Matlab 中,我有一个看起来像这样的向量:

0 0 1 1 0 0 0 1 1 0 0 0 0 1 1 1 0 0 0 0 1 0 1

我现在要做的是计算这个向量中 1 的个数。连续的 1 算作 1。此外,我还想计算 1 之间 0 的平均数和中位数。所以对于这个例子:

1秒:5

中位数 0:3.5

平均 0:3

我用蛮力方法解决了这个问题,即循环研究每个元素并检查前一个元素和下一个元素。但我确信必须有一个更快的解决方案。任何的想法?

4

3 回答 3

3

给定向量中的数据v

v = [ 0 0 1 1 0 0 0 1 1 0 0 0 0 1 1 1 0 0 0 0 1 0 1 ]; % data

计算如下:

w = [ 1 v 1 ]; % auxiliary vector
runs_zeros = find(diff(w)==1)-find(diff(w)==-1); % lenghts of runs of 0's

% Desired results:
number_ones = length(runs_zeros)-1+v(1)+v(end);
% For average and median, don't count first run if v(1) is 0,
% or last run if v(end) is 0:
average_runs_zeros = mean(runs_zeros(2-v(1):end-1+v(end))); 
median_runs_zeros = median(runs_zeros(2-v(1):end-1+v(end)));

这比@TryHard 的解决方案更快,因为它不需要转换为字符串

于 2013-09-08T14:23:32.243 回答
1

好的,所以这似乎有效

>> a=[0 0 1 1 0 0 0 1 1 0 0 0 0 1 1 1 0 0 0 0 1 0 1];
>> %Remove traling and leading zeros
>> y = a(find(a,1,'first'):find(a,1,'last'));
>> q = diff([0 a 0] == 1);
>> v = find(q == -1) - find(q == 1);
>> length(v) % Consecutive Ones
ans =

     5

>> q = diff([0 ~y 0] == 1);
>> v = find(q == -1) - find(q == 1);
>> v

v =

     3     4     4     1

>> median(v)

ans =

    3.5000

>> mean(v)

ans =

    3
于 2013-09-08T13:54:13.020 回答
0

你可以这样做:

dat=[0 0 1 1 0 0 0 1 1 0 0 0 0 1 1 1 0 0 0 0 1 0 1];
str = regexprep(num2str(dat),' ','');
[n1 istart1 iend1] = regexp(str,'[1]+','match','start','end');
[n0 istart0 iend0] = regexp(str(min(istart1):max(iend1)),'[0]+','match','start','end');

% number of strings of `1`s
length(n1)

% property of intercalated strings of `0`s
median([iend0-istart0+1])
mean([iend0-istart0+1])
于 2013-09-08T13:42:17.157 回答