CODEWITHSUNDEEP
pythonpython-3.xroundingcomplex-numbers
prav
prav

Reputation: 205

How to round up a complex number?

How can I round up a complex number (e.g. 1.9999999999999998-2j) as 2-2j?

When I tried using 

print(round(x,2))

it showed

Traceback (most recent call last):
  File "C:\Python34\FFT.py", line 22, in <module>
    print(round(x,2))
TypeError: type complex doesn't define __round__ method

Upvotes: 15

Views: 17684

Answers (5)

polarise
polarise

Reputation: 2413

Well, perhaps you can write your own _complex for local use? Here's an example:

class _complex(complex):
    def __round__(self, n=None):
        try:
            assert isinstance(n, int)
        except AssertionError:
            raise ValueError(f"n must be an integer: {type(n)}")
        if n is not None:
            return complex(round(self.real, n), round(self.imag, n))
        return self

And you use it like:

c = _complex(1, 2)
print(round(c, 4))

Very rough... may require some cleanup. I'm surprised this is not in the Python standard library.

Upvotes: 2

user14591479
user14591479

Reputation:

You could round the real part and imaginary part separately rather than combined. Like:

x=complex(1.9999999999999998,-2)
rounded_x=complex(round(x.real,2),round(x.imag,2))

Then you could print the rounded_x variable as a string(to avoid the brackets when printing). Hope this short answer helps the readers including the questioner.

Upvotes: 0

Evelyn
Evelyn

Reputation: 85

Id say the best way to do it is as such

x = (1.542334+32.5322j)
x = complex(round(x.real),round(x.imag))

if you don't want to repeat that every time you want to do it, you could put it in a function. 

def round_complex(x):
    return complex(round(x.real),round(x.imag))

Additional optional arguments can then be added to this, so if you only want to round one part for example, or if you only want to round to a certain number of decimal places on either the real or complex part

def round_complex(x, PlacesReal = 0, PlacesImag = 0, RoundImag = True, RoundReal = True):
     if RoundImag and not RoundReal:
         return complex(x.real,round(x.imag,PlacesImag))

     elif RoundReal and not RoundImag:
         return complex(round(x.real,PlacesReal),x.imag)

     else: #it would be a waste of space to make it do nothing if you set both to false, so it instead does what it would if both were true
         return complex(round(x.real,PlacesReal),round(x.imag,PlacesImag))

as the variables are auto set to true or 0, you don't need to input them unless you specifically want too. But they are handy to have

Upvotes: 3

schesis
schesis

Reputation: 59148

If all you want to do is represent the value rounded as shown, rather than modify the value itself, the following works:

>>> x=1.9999999999999998-2j
>>> print("{:g}".format(x))
2-2j

See: Format Specification Mini-Language.

Upvotes: 10

falsetru
falsetru

Reputation: 369134

Round real part and imaginary part separately and combine them:

>>> num = 1.9999999999999998-2j
>>> round(num.real, 2) + round(num.imag, 2) * 1j
(2-2j)

Upvotes: 13

Related Questions