in python can you code the square root symbol in python code example

Example: python square root

import math
a = input("what do you wnat to square root")
print(math.sqrt(a))