MATLAB矩阵元素条件函数定义及高效实现问题求解

需求说明 #

定义函数 y(x)，规则如下：

if x >= 0:     y = x^(1/2)
if x < 0:       y = NaN

尝试的无效代码 #

最初尝试的代码（未正确使用元素级运算符）：

y = @(x) (x>=0) * x.^(1/2) - NaN* (x<0); 
Test = y([-1,1,4,9])

修正运算符后仍无效的代码：

y = @(x) (x>=0) .* x^(1/2) - NaN.* (x<0); 
Test = y([-1,1,4,9])

期望输出 #

y([-1,1,4,9]) = [NaN, 1, 2, 3];

高效解决方案 #

方法1：向量化匿名函数（推荐） #

利用元素级运算和NaN的特性构造向量化匿名函数，避免循环或arrayfun，兼顾效率与简洁性：

y = @(x) (x >= 0) .* sqrt(x) + (x < 0) .* NaN;
% 测试
Test = y([-1, 1, 4, 9])

运行输出：

Test =
   NaN     1     2     3

原理：

• x >= 0生成逻辑数组，正数位置为1、负数为0，与sqrt(x)元素级相乘后，负数位置结果为0*复数=0，正数位置为正常平方根；
• x < 0生成的逻辑数组在负数位置为1、正数为0，与NaN元素级相乘后，负数位置为NaN、正数位置为0；
• 两部分相加后，负数位置0 + NaN = NaN，正数位置平方根 + 0 = 平方根，完全符合需求。

方法2：分步定义函数（更直观） #

如果不需要单行匿名函数，分步处理的代码可读性更强，同样是向量化操作，效率不受影响：

function y = my_sqrt(x)
    y = sqrt(x);
    y(x < 0) = NaN;
end
% 测试
Test = my_sqrt([-1, 1, 4, 9])

之前代码无效的原因 #

1. 幂运算错误：x^(1/2)是矩阵幂运算，仅适用于方阵；元素级开平方需用x.^(1/2)或sqrt(x)；
2. 运算逻辑疏漏：原代码用减法处理NaN，但0 - NaN虽能得到NaN，却因矩阵幂运算导致数组输入直接报错，修正元素级运算符后，改用加法逻辑会更简洁直观。

内容的提问来源于stack exchange，提问作者Romalpa Akzo