MATLAB, как использовать функцию FIND с оператором IF - PullRequest
0 голосов
/ 20 июня 2010

Этот вопрос относится к Как объединить эти данные в 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, если n равно n-1 (20 = 20) столбца 1, данные столбца 2 индекса n и n-1 объединены.

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

ИЗД.1021 *

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 голосов
/ 20 июня 2010

Насколько я могу судить, вам не нужно использовать find или if. Вам нужна только логическая индексация.

Если я правильно понимаю ваш вопрос, Амро прав, указывая, что ваш вывод не соответствует вашему описанию логики. Попробуйте следующий скрипт:

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]
0 голосов
/ 20 июня 2010

Нет необходимости в операторе 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
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...