Кубические корни комплексного числа в python

Question

Как рассчитать корни куба комплексного числа в python? В настоящее время я нашел единственный способ сделать это в соответствии с этим ответом, но я пытаюсь удалить все и все import утверждения из моего кода.

---

Текущий метод, воспроизведенный здесь:

import math
def cuberoot( z ):
    z = complex(z) 
    x = z.real
    y = z.imag
    mag = abs(z)
    arg = math.atan2(y,x)
    resMag = mag**(1./3)
    resArg = [ (arg+2*math.pi*n)/3. for n in range(1,4) ]
    return [  resMag*(math.cos(a) + math.sin(a)*1j) for a in resArg ]

Answer 1

Вычислите один и умножьте на кубические корни 1?

cuberoots_of_1 = 1, complex(-.5, .75**.5), complex(-.5, -.75**.5)

def cuberoot(z):
    cuberoot = complex(z)**(1/3)
    return [cuberoot * cr1 for cr1 in cuberoots_of_1]

Тест:

>>> cuberoot(-2 + 3j)
[(1.1532283040274223+1.0106429470939737j),
 (-1.4518566183526649+0.49340353410400484j),
 (0.2986283143252425-1.5040464811979786j)]

>>> for z in cuberoot(-2 + 3j):
        print(z**3, abs(z**3 - (-2 + 3j)))

(-1.999999999999999+3j) 1.1102230246251565e-15
(-1.999999999999999+3j) 1.1102230246251565e-15
(-1.9999999999999982+2.9999999999999996j) 1.831026719408895e-15

То же самое с вашим делом менее точно:

(-1.999999999999996+3.000000000000002j) 4.572178254219406e-15
(-1.9999999999999933+3.000000000000004j) 7.768388458966724e-15
(-1.9999999999999956+3.0000000000000013j) 4.636427468134552e-15

Answer 2

Вот подход, использующий вращение с коэффициентом i^(2/3).

f = 1j**(2/3)
def cube_roots(z):
  r = z**(1/3)
  return [r, -r*f, r*f**2]

Answer 3

Python встроенный complex может обрабатывать поиск одного root из коробки:

def cube_root(v):
    if not isinstance(v, complex):
        v = complex(v, 0)
    return v ** (1.0 / 3.0)

Примеры:

cube_root(-3)
(0.7211247851537043+1.2490247664834064j)

cube_root(complex(1, -2))
(1.2196165079717578-0.47171126778938893j)

Функция, которую вы воспроизвели выше один способ получить все три корня.