首页 > 解决方案 > 计算分段函数中的向量

问题描述

我想在这样的分段函数中评估向量 xdata:


dat=load('k2.txt');
xdata = dat(:,1);
ydata = dat(:,2);
n=length(xdata);
p0=[0.0821 6.6 0.4];
y(p0,xdata)

function y_hat = y(p,t)
    P=4.885;
    T0 = 134.27426;
    omega=2*pi/P;
    gamma1=0.3539 ;gamma2=0.2851;
    c1=0;c2=gamma1+2*gamma2;c3=0;c4=-gamma2;
    c0=1-c1-c2-c3-c4; 
    z= p(2).*((sin(omega.*(t-T0))).^2+((p(3)/p(2)).*cos(omega.*(t-T0))).^2).^(1/2);
    lambda1= 0; 
    lambda3=p(1).^2;

        if ((1-p(1)<z) & (z<1+p(1)))
            k1 = acos((1-p(1).^2 + z.^2)./(2*z));
            k0 = acos(((p(1)).^2+z.^2-1)./(2.*z.*p(1)));
            y_hat = 1-1./pi*(p(1).^2.*k0+k1-sqrt((4*z.^2-(1+z.^2-p(1).^2).^2)/4));

        end

        if (1+p(1)<=z)
            y_hat=1-lambda1; 

        end

        if (z<=1-p(1))
            y_hat=1-lambda3; 

        end
end

问题是代码没有输入任何 if 循环,也没有返回任何内容。也许原因是该函数试图一次满足所有向量的条件?我应该如何进行以便 y(p0,xdata) 返回一些东西?

顺便说一句,我需要这个,因为我必须为这样的数据拟合模型:


[theta] = lsqcurvefit(@y, p0, xdata, ydata);

标签: matlabdata-fittingmodel-fittingpiecewise

解决方案


你的代码不起作用,因为当你写这样的东西时:

if [1 3 -1] > 0
   %code
end

然后 "%code" 不会被评估,因为它正在检查条件是否适用于向量中的每个值。

假设您要定义Heaviside函数并评估它的向量。然后,您所做的是使用 for 循环:

x_vals = [-1 1 5];
heav(x_vals)
function y = heav(x)
    y = zeros(size(x));
    for i = 1:length(x)
        if x(i) >= 0
            y(i) = 1;
        else
            y(i) = 0;
        end
    end
end

推荐阅读