r - 复平面中的多个根与 R

Question

我一直在尝试找到一个函数来返回方程的所有复杂解，例如：

16^(1/4) = 2+i0,  -2+i0,  0+i2,  0-i2

就目前而言，如果我进入16^(1/4)控制台，它只返回 2。我可以为此编写一个函数，但我想知道在 R 中是否有一种简单的方法可以做到这一点。

Answer 1

你需要polyroot()：

polyroot(z = c(-16,0,0,0,1))
# [1]  0+2i -2-0i  0-2i  2+0i

其中z是“升序多项式系数向量”。

我在上面的例子中传递的向量z是这个等式的一个紧凑表示：

-16x^0 + 0x^1 + 0x^2 + 0x^3 + 1x^4 = 0

                          x^4 - 16 = 0

                               x^4 = 16

                                 x = 16^(1/4)

---

编辑：

如果polyroot's 的语法让您感到困扰，您只需编写一个包装函数，为您提供一个更好（如果不太通用）的界面：

nRoot <- function(x, root) {
    polyroot(c(-x, rep(0, root-1), 1))
}
nRoot(16, 4)
# [1]  0+2i -2-0i  0-2i  2+0i
nRoot(16, 8)
# [1]  1.000000+1.000000i -1.000000+1.000000i -1.000000-1.000000i
# [4]  1.000000-1.000000i  0.000000+1.414214i -1.414214-0.000000i
# [7]  0.000000-1.414214i  1.414214+0.000000i