MATLAB使用IF语句返回向量函数

Question

我正在调用一个像这样的向量的自编写函数'func'：

x_values=[0 1 2];
result=func(x_values);

问题是在这个函数中我有一个if语句来确定输出。如果我将此函数应用于标量，我没有问题，但如果我将它应用于数字向量，if语句不能完成他的工作。为什么？我该如何修理它？

function [y]=func(x)
if(x==0)
  y=0
else
  y=1./sin(x);
end
end

Answer 1

您需要分别处理零和非零条目。这很容易通过索引来实现：

function [y]=func(x)
xIsZero = x==0;

%# preassign y
y = x;

%# fill in values for y
y(xIsZero) = 0

y(~xIsZero) = 1./sin(x(~xIsZero))

Answer 2

答案在IF的帮助文档中解释：

  

当结果为非空时，求值表达式为true   包含所有非零元素（逻辑或实数）。除此以外，   表达式是假的。

结论：要么添加for循环，要么找到更好的方法来对表达式进行矢量化

Answer 3

您只想分配矢量的子集。例如：

function [y]=func(x)
    y=zeros(size(x));
    mask = x ~= 0;
    y(mask) = 1 ./ sin(x(mask));
end