en python como se trabajan las raices cuadradas ?

Sí, exactamente con

# importas math 
import math  
  
# imprimes la raíz cuadrada de 9
print(math.sqrt(9))  

Y aquí puedes leer un poco más al respecto: https://www.geeksforgeeks.org/python-math-function-sqrt/

También lo puedes hacer elevándolo a la 0.5 (para elevar un numero al cuadrado se usa ** ) Ej: 4**0.5 = 2

Para las raíces se pueden hacer uso de los exponentes inversos

4**(1/2) //raíz cuadrada de 4
8**(1/3) //raíz cúbica de 8