数学建模9.2代码记录

clear; clc
M1=dlmread('data0902.txt');  % 读数据文件 
n=size(M1,1);%样本数
Y=M1(:,2); % 薪金
for i=1:n
    if M1(i,5)==1
        x3(i)=1;x4(i)=0;
    elseif M1(i,5)==2
        x3(i)=0;x4(i)=1;
    else 
        x3(i)=0;x4(i)=0;
    end
end
X=[ones(46,1),M1(:,[3,4]),x3',x4'];
[b,bint,r,rint,ss]=regress(Y,X);
plot(X(:,2),r,'+')
% 定义Z 代表管理-教育组合
for i=1:n
    if X(i,3)==0 && M1(i,5)==1
        Z(i)=1;
    elseif X(i,3)==1 && M1(i,5)==1
        Z(i)=2;
    elseif X(i,3)==0 && M1(i,5)==2
        Z(i)=3;
    elseif X(i,3)==1 && M1(i,5)==2
        Z(i)=4;
    elseif X(i,3)==0 && M1(i,5)==3
        Z(i)=5;
    else
        Z(i)=6;
    end
end
plot(Z,r,'+')
X2=[X,X(:,3).*X(:,4),X(:,3).*X(:,5)];
[b2,bint2,r2,rint2,ss2]=regress(Y,X2);
plot(X(:,2),r2,'+')
plot(Z,r2,'+')
% X2中删去异常值，即第33行
X2(33,:)=[];Y(33,:)=[];