自创matlab函数，求复数的n次方根并输出所有复数解

学过复变函数的同学会知道，一个复数的n次方根（n为正整数）有n个。比如-1的二次方根有两个，分别是-i和+i。但是如果用matlab执行sqrt(-1)或者(-1)^0.5,只能得到一个根：

(-1)^0.5
ans =
   0.0000 + 1.0000i

为了解决这个问题，我自己写了一段matlab函数，如下：

%求复数的正整数次方根，并且返回所有复数根。
% X可以是标量，也可以是向量。
% n必须是正整数。
% Y是L*n的矩阵，其中L是X的长度。
function Y=complex_sqrt(X,n)
[~,b]=size(X);
if b>1
    X=X.';%进行转置但复数元素的值不会改变，使用"'"转置会造成复数元素的值改变
end
pre_angle=angle(X)+2*pi*(angle(X)<0);%angle的输出范围是[-pi,pi]，我们希望的范围是[0,2pi)
complex_angle=(pre_angle+(0:2*pi:2*(n-1)*pi))/n;
Y=repmat(abs(X).^(1/n),[1,n]).*(cos(complex_angle)+1i*sin(complex_angle));

这个函数能求X的n次方根，其中n为正整数，X是复数，可以是标量也可以是向量，但还不支持矩阵。如果X是标量，则输出Y是n个元素的行向量，表示X的所有n次方根；如果X是向量，则输出L*n的矩阵，L是X的长度，每行代表X中对应元素的所有n次方根。

例如，求i的二次方根：

complex_sqrt(1i,2)
ans =
   0.7071 + 0.7071i  -0.7071 - 0.7071i

再例如，求向量X中每个元素的三次方根：

 X=[4*sqrt(2)*(-1+1i),-i]
X =
  -5.6569 + 5.6569i   0.0000 - 1.0000i
>> complex_sqrt(X,3)
ans =
   1.4142 + 1.4142i  -1.9319 + 0.5176i   0.5176 - 1.9319i
   0.0000 + 1.0000i  -0.8660 - 0.5000i   0.8660 - 0.5000i

如果有错误，欢迎在评论区指正