MATLAB 2 – Python – Fórum – Programujte.com

Ahoj, chtěla bych se zeptat jestli by někdo vytvořil program (funkci) na odmocňování čísel tak aby to vypisovalo komplexní čísla. Nevím si stím rady a ráda bych se naučila jak to psát. Díky moc    

z = a + ib = r * e^(i * alfa), kde r = sqrt(a^2 + b^2) a alfa = atan(a,b) ale taky ze pre kazde a a b v R je definovany na [0, 2pi). Uz len staci odmocnit: (z)^(1/2) = (r  e^(i alfa))^(1/2) = r^(1/2) e^(i alfa/2).

V pythone by to vyzeralo asi takto:
 

def atan(x, y):
	"""Vypocitas uhol komplexneho cisla v komplexnej rovine"""

def abs(x, y):
	"""Vypocitas vzdialenost od 0"""

def sqrt(x, y):
	"""Vypocitas odmocninu"""
	r = abs(x, y)
	alfa = atan(x, y)
	sqrt_alfa = alfa/2
	sqrt_r = sqrt(r)
	return (sqrt_r * cos(sqrt_alfa), sqrt_r * sin(sqrt_alfa))

# cos a sin su standardne funkcie cosinus a sinus, btw. v pythone je
# definovana funkcia abs - nie je treba ju pisat
# to iste atan(a,b)
# a aj sqrt(z: complex)

x je realna cast a y je imag v pythone su definovane komplexne cisla aj operacie (funkcie) nad tou mnozinou a v matlabe zrejme tiez