此问题与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);
答案 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]