Cubic root of the negative number on python

by

A simple use of De Moivre’s formula, is sufficient to show that the cube root of a value, regardless of sign, is a multi-valued function. That means, for any input value, there will be three solutions. Most of the solutions presented to far only return the principle root. A solution that returns all valid roots, and explicitly tests for non-complex special cases, is shown below.

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

Edit: As requested, in cases where it is inappropriate to have dependency on numpy, the following code does the same thing.

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 ]

More Related Contents:

  1. Fastest way to list all primes below N
  2. Python math is wrong [duplicate]
  3. Check if a number is a perfect square
  4. Convert a number range to another range, maintaining ratio
  5. How can I convert radians to degrees with Python?
  6. C and Python – different behaviour of the modulo (%) operation
  7. How to make scipy.interpolate give an extrapolated result beyond the input range?
  8. Round float to x decimals?
  9. Math operations from string [duplicate]
  10. Difference between the built-in pow() and math.pow() for floats, in Python?
  11. Python: sort function breaks in the presence of nan
  12. How do I find the distance between two points? [closed]
  13. How to get all combination of n binary value? [duplicate]
  14. scipy is not optimizing and returns “Desired error not necessarily achieved due to precision loss”
  15. Linear regression of arrays containing NANs in Python/Numpy
  16. how can i show an irrational number to 100 decimal places in python?
  17. Is cube root integer?
  18. Multiple variables in SciPy’s optimize.minimize
  19. Rotate point about another point in degrees python
  20. Python floating-point math is wrong [duplicate]
  21. Equivalent of `polyfit` for a 2D polynomial in Python
  22. Computing Standard Deviation in a stream
  23. Get all numbers that add up to a number
  24. How to pad a string with leading zeros in Python 3 [duplicate]
  25. How to perform bilinear interpolation in Python
  26. Getting the subsets of a set in Python
  27. Python cos(90) and cos(270) not 0
  28. counting combinations and permutations efficiently
  29. Convert Latitude and Longitude to point in 3D space
  30. How do I get a decimal value when using the division operator in Python?

Leave a Comment