python - Trying to figure out python square root

Question

How would i do the following:

Write a function compmag(r,m) to compute, and return, the magnitude of a complex number. It should take the real component of the number as parameter r and the imaginary component as m. Remember that |r + mi| = \sqrt{r^2 + m^2}.

Does Python have a square root function? How would you find it?

Answer 1

您可以使用math模块或cmath模块。

>>> import math
>>> math.sqrt(4)
2
>>> import cmath
>>> cmath.sqrt(4 + 3j)
(2.1213203435596424+0.7071067811865476j)

请记住，普通math模块仅适用于普通数字。cmath如果要使用复数，则需要使用该模块。它们提供完全相同的功能，因此可以互换。唯一的区别是它cmath有点慢，因为它需要处理复数。

Answer 2

在math模块中。几乎所有与数学相关的函数都可以在那里找到。

import math
math.sqrt(2.0)

或者：

from math import sqrt
sqrt(2.0)

Answer 3

我会使用该cmath模块。math如果您使用复数，该模块会报错。例如，

import cmath
x = 1. + 1j
mod, ang = cmath.polar(x)
mod
1.4142135623730951

Answer 4

首先，导入数学：

import math

然后，编写 math.sqrt 函数：

print math.sqrt(Your number)