Какая функция нужна для корня числа в Python?

Для извлечения корня из числа лучше всего подойдет функция sqrt() из модуля math:

import math

root = math.sqrt(4)
# функция возвращает число с плавающей точкой float
print(root) #=> 2.0