MATLAB如何在IF语句中使用FIND函数

时间:2010-06-20 17:04:17

标签: matlab function find if-statement

此问题与How can I merge this data in MATLAB?

有关

如何将FIND函数与IF语句一起使用?例如如果我有这个数据:

20  10  1
20  11  1
20  15  1
23  10  1
23  10  1
23  12  0

规则1: 第3列的数据必须为1.

规则2: 如果n是第1列的当前索引,如果第1列n等于n-1(20 = 20),则索引n和n-1 的第2列数据为合并。

20  21  0
20  15  0
20  0   0
23  20  0
23  0   0
23  12  0

已编辑:

fid=fopen('data.txt');
A=textscan(fid,'%f%f%f');
fclose(fid);
in = cell2mat(A)'; %'# fix for SO formatting - Jonas

%# rows2mergeFrom are the rows where the second row equals the first row of col 1
%# and where the third column is 1. rows2mergeInto indicates the rows from which  the
%# values of the following rows should be added
rows2mergeFrom = find(in(2:end,1) == in(1:end-1,1) & in(2:end,3) == 1) + 1;

out = in;
out(rows2mergeFrom-1,2) = out(rows2mergeFrom-1,2) + out(rows2mergeFrom,2);

%# data that has been shifted up gets subtracted from the 'rows2mergeFrom'-rows
out(rows2mergeFrom,2) = out(rows2mergeFrom,2) - in(rows2mergeFrom,2);

%# remove the ones in the third column from any row that has been involved in a 
%# merge operation
out([rows2mergeFrom;rows2mergeFrom-1],3) = 0

fid = fopen('newData.txt','wt');  
format short g;
fprintf(fid,'%g\t %g\t %g\n',out);  %'# Write the data to the file
fclose(fid);  

2 个答案:

答案 0 :(得分:0)

不需要if语句。你需要的是一个逻辑数组,它允许find提取要合并的行的行索引。

已更新以符合新规则

in = [20  10  1
20  11  1
20  15  1
23  10  1
23  10  1
23  12  0];

%# rows2mergeFrom are the rows where the second row equals the first row of col 1
%# and where the third column is 1. rows2mergeInto indicates the rows from which  the
%# values of the following rows should be added
rows2mergeFrom = find(in(2:end,1) == in(1:end-1,1) & in(2:end,3) == 1) + 1;

out = in;
out(rows2mergeFrom-1,2) = out(rows2mergeFrom-1,2) + out(rows2mergeFrom,2);

%# data that has been shifted up gets subtracted from the 'rows2mergeFrom'-rows
out(rows2mergeFrom,2) = out(rows2mergeFrom,2) - in(rows2mergeFrom,2);

%# remove the ones in the third column from any row that has been involved in a 
%# merge operation
out([rows2mergeFrom;rows2mergeFrom-1],3) = 0

out =
20    21     0
20    15     0
20     0     0
23    20     0
23     0     0
23    12     0

答案 1 :(得分:0)

据我所知,你不需要使用find或if。您只需要逻辑索引。

如果我理解你的问题,Amro指出你的输出与你的逻辑描述不符。请尝试以下脚本:

m = [20  10  0; ...
     20  11  0; ...
     20  15  1; ...
     23  10  1; ...
     23  10  1];
merge = (m(1:end-1, 1) == m(2:end, 1)) & (m(2:end,3) == 1)
mnew = m(2:end, :);
madd = m(1:end-1,2) + m(2:end,2);
mnew(merge, 2) = madd(merge);
mnew = [m(1,:); mnew]